diff --git a/funasr/models/e2e_asr_paraformer.py b/funasr/models/e2e_asr_paraformer.py index 5a1a29be8..e157454e4 100644 --- a/funasr/models/e2e_asr_paraformer.py +++ b/funasr/models/e2e_asr_paraformer.py @@ -2107,7 +2107,7 @@ class ContextualParaformer(Paraformer): return loss_att, acc_att, cer_att, wer_att, loss_pre - def cal_decoder_with_predictor(self, encoder_out, encoder_out_lens, sematic_embeds, ys_pad_lens, hw_list=None): + def cal_decoder_with_predictor(self, encoder_out, encoder_out_lens, sematic_embeds, ys_pad_lens, hw_list=None, clas_scale=1.0): if hw_list is None: # default hotword list hw_list = [torch.Tensor([self.sos]).long().to(encoder_out.device)] # empty hotword list