From 9d48230c4f8f25bf88c5d6105f97370a36c9cf43 Mon Sep 17 00:00:00 2001
From: zhifu gao <zhifu.gzf@alibaba-inc.com>
Date: 星期一, 11 三月 2024 10:48:50 +0800
Subject: [PATCH] export onnx (#1457)
---
funasr/models/paraformer/decoder.py | 5 +++--
1 files changed, 3 insertions(+), 2 deletions(-)
diff --git a/funasr/models/paraformer/decoder.py b/funasr/models/paraformer/decoder.py
index ce018f4..572a34a 100644
--- a/funasr/models/paraformer/decoder.py
+++ b/funasr/models/paraformer/decoder.py
@@ -635,8 +635,9 @@
else:
self.make_pad_mask = sequence_mask(max_seq_len, flip=False)
- from funasr.models.sanm.multihead_att import MultiHeadedAttentionSANMDecoderExport
- from funasr.models.sanm.multihead_att import MultiHeadedAttentionCrossAttExport
+ from funasr.models.sanm.attention import MultiHeadedAttentionSANMDecoderExport
+ from funasr.models.sanm.attention import MultiHeadedAttentionCrossAttExport
+
for i, d in enumerate(self.model.decoders):
if isinstance(d.self_attn, MultiHeadedAttentionSANMDecoder):
--
Gitblit v1.9.1