From 85e7f2225cbad95a8464fe247e19e987ece7951b Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E5=98=89=E6=B8=8A?= Date: Mon, 15 May 2023 14:02:13 +0800 Subject: [PATCH] update repo --- funasr/models/e2e_asr_paraformer.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/funasr/models/e2e_asr_paraformer.py b/funasr/models/e2e_asr_paraformer.py index ef8c0ca2e..cc6f54ba0 100644 --- a/funasr/models/e2e_asr_paraformer.py +++ b/funasr/models/e2e_asr_paraformer.py @@ -906,9 +906,9 @@ class ParaformerBert(Paraformer): self.step_cur += 1 # for data-parallel text = text[:, : text_lengths.max()] - speech = speech[:, :speech_lengths.max(), :] + speech = speech[:, :speech_lengths.max()] if embed is not None: - embed = embed[:, :embed_lengths.max(), :] + embed = embed[:, :embed_lengths.max()] # 1. Encoder encoder_out, encoder_out_lens = self.encode(speech, speech_lengths)