From 4a33f08defa92ac6c9c757dbde06d29d183bf203 Mon Sep 17 00:00:00 2001
From: zhaomingwork <61895407+zhaomingwork@users.noreply.github.com>
Date: 星期二, 13 六月 2023 10:54:23 +0800
Subject: [PATCH] keep blank for english asr result (#623)

---
 funasr/export/models/modules/decoder_layer.py |    1 +
 1 files changed, 1 insertions(+), 0 deletions(-)

diff --git a/funasr/export/models/modules/decoder_layer.py b/funasr/export/models/modules/decoder_layer.py
index f539452..9a464a4 100644
--- a/funasr/export/models/modules/decoder_layer.py
+++ b/funasr/export/models/modules/decoder_layer.py
@@ -54,6 +54,7 @@
     
     def forward(self, tgt, tgt_mask, memory, memory_mask, cache=None):
         residual = tgt
+        tgt = self.norm1(tgt)
         tgt_q = tgt
         tgt_q_mask = tgt_mask
         x = residual + self.self_attn(tgt_q, tgt, tgt, tgt_q_mask)

--
Gitblit v1.9.1