modelscope/FunASR
1
0
Fork 0
mirror of https://github.com/modelscope/FunASR.git synced 2026-10-12 02:19:18 +00:00
Code Issues 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
Download Patch File Download Diff File Expand all files Collapse all files
runtime/python/libtorch/funasr_torch/sensevoice_bin.py
+3
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
runtime/python/onnxruntime/funasr_onnx/sensevoice_bin.py
+3
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