diff --git a/funasr/models/llm_asr/tts_models/e2e_model.py b/funasr/models/llm_asr/tts_models/e2e_model.py index 6ee922b96..59c65eed8 100644 --- a/funasr/models/llm_asr/tts_models/e2e_model.py +++ b/funasr/models/llm_asr/tts_models/e2e_model.py @@ -1004,6 +1004,8 @@ class UCTDXvecSlotModel(UpsampleCtcTokenDiffModel): # forward FM model set_all_random_seed(0) + if cur_token.shape[1] < 1: + return None, None feat = self.fm_model.inference( cur_token, cur_token_len, xvec, xvec_lengths,