Explorar el Código

update infer func for old contextual

shixian.shi hace 2 años
padre
commit
5690a8bcb3
Se han modificado 1 ficheros con 1 adiciones y 1 borrados
  1. 1 1
      funasr/models/e2e_asr_paraformer.py

+ 1 - 1
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