From ad75a464d8f538b74eb1f11ba6bd3409901136dc Mon Sep 17 00:00:00 2001
From: 仁迷 <haoneng.lhn@alibaba-inc.com>
Date: 星期一, 09 一月 2023 19:34:52 +0800
Subject: [PATCH] update text postprocess

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

diff --git a/funasr/models/e2e_asr_paraformer.py b/funasr/models/e2e_asr_paraformer.py
index 89f7cf0..3f8359d 100644
--- a/funasr/models/e2e_asr_paraformer.py
+++ b/funasr/models/e2e_asr_paraformer.py
@@ -493,7 +493,7 @@
 	def sampler(self, encoder_out, encoder_out_lens, ys_pad, ys_pad_lens, pre_acoustic_embeds):
 
 		tgt_mask = (~make_pad_mask(ys_pad_lens, maxlen=ys_pad_lens.max())[:, :, None]).to(ys_pad.device)
-		ys_pad *= tgt_mask[:, :, 0]
+		ys_pad = ys_pad * tgt_mask[:, :, 0]
 		ys_pad_embed = self.decoder.embed(ys_pad)
 		with torch.no_grad():
 			decoder_outs = self.decoder(

--
Gitblit v1.9.1