From 8b3e0e52c50f0bf27cc2eb5a44b428e44f48ae62 Mon Sep 17 00:00:00 2001
From: jmwang66 <wangjiaming.wjm@alibaba-inc.com>
Date: 星期一, 26 十二月 2022 22:55:56 +0800
Subject: [PATCH] update github.io page

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