speech_recognition/FunASR
1
0
Fork 0
mirror of https://github.com/modelscope/FunASR synced 2025-09-15 14:48:36 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

update repo

Browse Source
This commit is contained in:
嘉渊 2023-05-15 14:02:13 +08:00
parent eb43576ed0
commit 85e7f2225c
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
funasr/models/e2e_asr_paraformer.py
View File

@ -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)