diff --git a/funasr/models/llm_asr/model.py b/funasr/models/llm_asr/model.py index 55d43c796..5956db04d 100644 --- a/funasr/models/llm_asr/model.py +++ b/funasr/models/llm_asr/model.py @@ -2377,7 +2377,7 @@ class LLMASR5(nn.Module): if torch.any(top_ids == (self.codebook_size + self.ad_sos_eos)): hit_eos = True - out_tokens = out_tokens[:, :out_token_len+1, :] + out_tokens = out_tokens[:, :out_token_len, :] break out_tokens[0, out_token_len, 0] = top_ids[0]