EngineX-Iluvatar/enginex_bi_series-sherpa-onnx
Archived
16
0
Fork 0
Code Issues Pull Requests Actions Projects Releases Wiki Activity

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

Browse Source 
* 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
This commit is contained in:
Fangjun Kuang
2024-06-24 21:52:54 +08:00
committed by GitHub
parent 1f95bff719
commit a3bac19c54
4 changed files with 6 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
.github/workflows/test-build-wheel.yaml vendored
View File

@@ -132,7 +132,7 @@ jobs:
export PATH=/c/hostedtoolcache/windows/Python/3.9.13/x64/bin:$PATH
export PATH=/c/hostedtoolcache/windows/Python/3.10.11/x64/bin:$PATH
export PATH=/c/hostedtoolcache/windows/Python/3.11.9/x64/bin:$PATH
export PATH=/c/hostedtoolcache/windows/Python/3.12.3/x64/bin:$PATH
export PATH=/c/hostedtoolcache/windows/Python/3.12.4/x64/bin:$PATH
which sherpa-onnx
sherpa-onnx --help

2
.github/workflows/test-pip-install.yaml vendored
View File

@@ -107,7 +107,7 @@ jobs:
export PATH=/c/hostedtoolcache/windows/Python/3.9.13/x64/bin:$PATH
export PATH=/c/hostedtoolcache/windows/Python/3.10.11/x64/bin:$PATH
export PATH=/c/hostedtoolcache/windows/Python/3.11.9/x64/bin:$PATH
export PATH=/c/hostedtoolcache/windows/Python/3.12.3/x64/bin:$PATH
export PATH=/c/hostedtoolcache/windows/Python/3.12.4/x64/bin:$PATH
sherpa-onnx --help
sherpa-onnx-keyword-spotter --help

2
scripts/wenet/test-onnx-streaming.py
View File

@@ -143,7 +143,7 @@ def main():
(model.chunk_size - 1) * model.subsampling_factor + model.right_context + 1
)
chunk_length = int(chunk_length)
chunk_shift = int(model.required_cache_size)
chunk_shift = int(model.chunk_size * model.subsampling_factor)
print(chunk_length, chunk_shift)
num_frames = x.shape[0]

4
sherpa-onnx/csrc/online-wenet-ctc-model.cc
View File

@@ -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_; }