Sfoglia il codice sorgente

update infer func for old contextual

shixian.shi 2 anni fa
parent
commit
5690a8bcb3
1 ha cambiato i file con 1 aggiunte e 1 eliminazioni
  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