From c9f1b4e8a2e903f74de20d019e70307c26e93c3e Mon Sep 17 00:00:00 2001 From: "shixian.shi" Date: Thu, 23 Nov 2023 20:39:52 +0800 Subject: [PATCH] update --- funasr/models/e2e_asr_contextual_paraformer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/funasr/models/e2e_asr_contextual_paraformer.py b/funasr/models/e2e_asr_contextual_paraformer.py index 64e0f8d03..171a6c60a 100644 --- a/funasr/models/e2e_asr_contextual_paraformer.py +++ b/funasr/models/e2e_asr_contextual_paraformer.py @@ -134,7 +134,7 @@ class NeatContextualParaformer(Paraformer): text_lengths: torch.Tensor, hotword_pad: torch.Tensor, hotword_lengths: torch.Tensor, - ideal_attn: torch.Tensor, + dha_pad: torch.Tensor, ) -> Tuple[torch.Tensor, Dict[str, torch.Tensor], torch.Tensor]: """Frontend + Encoder + Decoder + Calc loss