From 7c4cfcfab4f4270c344bd07e1622eb84f85100e3 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E6=B8=B8=E9=9B=81?= Date: Mon, 17 Jun 2024 14:38:22 +0800 Subject: [PATCH] decoding --- funasr/models/llm_asr/model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/funasr/models/llm_asr/model.py b/funasr/models/llm_asr/model.py index 03a2c08b0..ea039a5f4 100644 --- a/funasr/models/llm_asr/model.py +++ b/funasr/models/llm_asr/model.py @@ -1018,7 +1018,7 @@ class LLMASR4(nn.Module): for turn_id in range(fbank_beg.shape[1]): fbank_beg_idx = fbank_beg[batch_idx, turn_id].item() - if fbank_beg[batch_idx, turn_id] > 0: + if fbank_beg_idx > 0: speech_token_len = fake_token_len[batch_idx, turn_id] speech_token = encoder_out[speech_idx, :speech_token_len, :]