From 08d39eae7d9886d37ab5d7ae2f579b85b6491e50 Mon Sep 17 00:00:00 2001
From: onlybetheone <iriszhangchong@gmail.com>
Date: 星期四, 22 十二月 2022 17:21:52 +0800
Subject: [PATCH] FunTextProcessing

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