From a9e857e45250b16af60d5fe3efcd06e685f6506a Mon Sep 17 00:00:00 2001
From: lzr265946 <lzr265946@alibaba-inc.com>
Date: 星期六, 03 十二月 2022 16:39:38 +0800
Subject: [PATCH] update funasr 0.1.3

---
 funasr/models/e2e_asr_paraformer.py |    5 ++---
 1 files changed, 2 insertions(+), 3 deletions(-)

diff --git a/funasr/models/e2e_asr_paraformer.py b/funasr/models/e2e_asr_paraformer.py
index 5ea28f3..89f7cf0 100644
--- a/funasr/models/e2e_asr_paraformer.py
+++ b/funasr/models/e2e_asr_paraformer.py
@@ -330,9 +330,10 @@
 
 	def cal_decoder_with_predictor(self, encoder_out, encoder_out_lens, sematic_embeds, ys_pad_lens):
 
-		decoder_out, _ = self.decoder(
+		decoder_outs = self.decoder(
 			encoder_out, encoder_out_lens, sematic_embeds, ys_pad_lens
 		)
+		decoder_out = decoder_outs[0]
 		decoder_out = torch.log_softmax(decoder_out, dim=-1)
 		return decoder_out, ys_pad_lens
 
@@ -553,7 +554,6 @@
 		postencoder: Optional[AbsPostEncoder],
 		decoder: AbsDecoder,
 		ctc: CTC,
-		joint_network: Optional[torch.nn.Module],
 		ctc_weight: float = 0.5,
 		interctc_weight: float = 0.0,
 		ignore_id: int = -1,
@@ -590,7 +590,6 @@
 		postencoder=postencoder,
 		decoder=decoder,
 		ctc=ctc,
-		joint_network=joint_network,
 		ctc_weight=ctc_weight,
 		interctc_weight=interctc_weight,
 		ignore_id=ignore_id,

--
Gitblit v1.9.1