speech_recognition/FunASR
1
0
Fork 0
mirror of https://github.com/modelscope/FunASR synced 2025-09-15 14:48:36 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

bug fix for onnx infer

Browse Source
This commit is contained in:
维石 2024-07-26 11:40:23 +08:00
parent af1c0d0e30
commit 874ed3aaac
2 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
runtime/python/libtorch/funasr_torch/sensevoice_bin.py
View File

@ -158,6 +158,9 @@ class SenseVoiceSmall:
feats, feats_len = self.extract_feat(waveform_list[beg_idx:end_idx])
_language_list = language_list[beg_idx:end_idx]
_textnorm_list = textnorm_list[beg_idx:end_idx]
if not len(_language_list):
_language_list = [language_list[0]]
_textnorm_list = [textnorm_list[0]]
B = feats.shape[0]
if len(_language_list) == 1 and B != 1:
_language_list = _language_list * B

3
runtime/python/onnxruntime/funasr_onnx/sensevoice_bin.py
View File

@ -165,6 +165,9 @@ class SenseVoiceSmall:
feats, feats_len = self.extract_feat(waveform_list[beg_idx:end_idx])
_language_list = language_list[beg_idx:end_idx]
_textnorm_list = textnorm_list[beg_idx:end_idx]
if not len(_language_list):
_language_list = [language_list[0]]
_textnorm_list = [textnorm_list[0]]
B = feats.shape[0]
if len(_language_list) == 1 and B != 1:
_language_list = _language_list * B