From 0ca6876f5814aeb6802e050223427ff510a4a160 Mon Sep 17 00:00:00 2001 From: "haoneng.lhn" Date: Mon, 3 Apr 2023 14:15:51 +0800 Subject: [PATCH] update --- funasr/models/decoder/sanm_decoder.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/funasr/models/decoder/sanm_decoder.py b/funasr/models/decoder/sanm_decoder.py index 3bfcffc3f..a36d95e77 100644 --- a/funasr/models/decoder/sanm_decoder.py +++ b/funasr/models/decoder/sanm_decoder.py @@ -94,7 +94,7 @@ class DecoderLayerSANM(nn.Module): if self.self_attn: if self.normalize_before: tgt = self.norm2(tgt) - x, _ = self.self_attn(tgt, tgt_mask) + x, cache = self.self_attn(tgt, tgt_mask, cache=cache) x = residual + self.dropout(x) if self.src_attn is not None: @@ -104,7 +104,6 @@ class DecoderLayerSANM(nn.Module): x = residual + self.dropout(self.src_attn(x, memory, memory_mask)) - return x, tgt_mask, memory, memory_mask, cache def forward_chunk(self, tgt, tgt_mask, memory, memory_mask=None, cache=None):