fix a bug for wenet streaming model. (#1054)

* fix a bug for wenet streaming model. The chunk shift was wrong. See https://github.com/wenet-e2e/wenet/blob/main/runtime/core/decoder/asr_model.cc#L15 and https://github.com/wenet-e2e/wenet/blob/main/runtime/core/decoder/asr_model.cc#L28
2024-06-24 21:52:54 +08:00
parent 1f95bff719
commit a3bac19c54
4 changed files with 6 additions and 4 deletions
--- a/sherpa-onnx/csrc/online-wenet-ctc-model.cc
+++ b/sherpa-onnx/csrc/online-wenet-ctc-model.cc
@@ -97,7 +97,9 @@ class OnlineWenetCtcModel::Impl {
           right_context_ + 1;
  }

-  int32_t ChunkShift() const { return required_cache_size_; }
+  int32_t ChunkShift() const {
+    return config_.wenet_ctc.chunk_size * subsampling_factor_;
+  }

  OrtAllocator *Allocator() const { return allocator_; }