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

[fix] Fix a bug in seaco_paraformer model "inference" function (#1639)

Browse Source
This commit is contained in:
querryton 2024-04-20 16:07:13 +08:00 committed by GitHub
parent dee1354d0d
commit 01df8f330c
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
funasr/models/seaco_paraformer/model.py
View File

@ -350,7 +350,7 @@ class SeacoParaformer(BiCifParaformer, Paraformer):
pre_acoustic_embeds, pre_token_length = predictor_outs[0], predictor_outs[1] pre_acoustic_embeds, pre_token_length = predictor_outs[0], predictor_outs[1]
pre_token_length = pre_token_length.round().long() pre_token_length = pre_token_length.round().long()
if torch.max(pre_token_length) < 1: if torch.max(pre_token_length) < 1:
return [] return ([],)
decoder_out = self._seaco_decode_with_ASF(encoder_out, decoder_out = self._seaco_decode_with_ASF(encoder_out,
encoder_out_lens, encoder_out_lens,