diff --git a/runtime/python/libtorch/funasr_torch/sensevoice_bin.py b/runtime/python/libtorch/funasr_torch/sensevoice_bin.py index 4f793406d..d4444e7df 100644 --- a/runtime/python/libtorch/funasr_torch/sensevoice_bin.py +++ b/runtime/python/libtorch/funasr_torch/sensevoice_bin.py @@ -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 diff --git a/runtime/python/onnxruntime/funasr_onnx/sensevoice_bin.py b/runtime/python/onnxruntime/funasr_onnx/sensevoice_bin.py index 41f9042a6..37c98a8d0 100644 --- a/runtime/python/onnxruntime/funasr_onnx/sensevoice_bin.py +++ b/runtime/python/onnxruntime/funasr_onnx/sensevoice_bin.py @@ -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