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 code

Browse Source
This commit is contained in:
shixian.shi 2024-03-08 11:37:46 +08:00
parent 74f4f7b4c7
commit 7498bd7388
1 changed files with 1 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -175,11 +175,7 @@ class SeacoParaformer(BiCifParaformer, Paraformer):
encoder_out_mask = (~make_pad_mask(encoder_out_lens, maxlen=encoder_out.size(1))[:, None, :]).to( encoder_out_mask = (~make_pad_mask(encoder_out_lens, maxlen=encoder_out.size(1))[:, None, :]).to(
encoder_out.device) encoder_out.device)
predictor_outs = self.predictor(encoder_out, None, encoder_out_mask, ignore_id=self.ignore_id) predictor_outs = self.predictor(encoder_out, None, encoder_out_mask, ignore_id=self.ignore_id)
if len(predictor_outs) == 4: return predictor_outs[:4]
pre_acoustic_embeds, pre_token_length, alphas, pre_peak_index = predictor_outs
else:
pre_acoustic_embeds, pre_token_length, alphas, pre_peak_index, pre_token_length2 = predictor_outs
return pre_acoustic_embeds, pre_token_length, alphas, pre_peak_index
def _calc_seaco_loss( def _calc_seaco_loss(
self, self,