Add HLG decoding for streaming CTC models (#731)

2024-04-03 21:31:42 +08:00
parent f8832cb5f2
commit db67e00c77
28 changed files with 668 additions and 82 deletions
--- a/sherpa-onnx/csrc/online-ctc-greedy-search-decoder.cc
+++ b/sherpa-onnx/csrc/online-ctc-greedy-search-decoder.cc
@@ -13,7 +13,8 @@
 namespace sherpa_onnx {

 void OnlineCtcGreedySearchDecoder::Decode(
-    Ort::Value log_probs, std::vector<OnlineCtcDecoderResult> *results) {
+    Ort::Value log_probs, std::vector<OnlineCtcDecoderResult> *results,
+    OnlineStream ** /*ss=nullptr*/, int32_t /*n = 0*/) {
  std::vector<int64_t> log_probs_shape =
      log_probs.GetTensorTypeAndShapeInfo().GetShape();