From 53fccccb24d15d788919d91c8c2b06a115ddacf3 Mon Sep 17 00:00:00 2001 From: 夜雨飘零 <yeyupiaoling@foxmail.com> Date: 星期二, 19 十二月 2023 15:39:39 +0800 Subject: [PATCH] 修改cnn为合理的包名 --- funasr/export/models/decoder/contextual_decoder.py | 6 +++--- 1 files changed, 3 insertions(+), 3 deletions(-) diff --git a/funasr/export/models/decoder/contextual_decoder.py b/funasr/export/models/decoder/contextual_decoder.py index 4e11b5d..c6f83d0 100644 --- a/funasr/export/models/decoder/contextual_decoder.py +++ b/funasr/export/models/decoder/contextual_decoder.py @@ -4,11 +4,11 @@ from funasr.export.utils.torch_function import MakePadMask from funasr.export.utils.torch_function import sequence_mask -from funasr.modules.attention import MultiHeadedAttentionSANMDecoder +from funasr.models.transformer.attention import MultiHeadedAttentionSANMDecoder from funasr.export.models.modules.multihead_att import MultiHeadedAttentionSANMDecoder as MultiHeadedAttentionSANMDecoder_export -from funasr.modules.attention import MultiHeadedAttentionCrossAtt +from funasr.models.transformer.attention import MultiHeadedAttentionCrossAtt from funasr.export.models.modules.multihead_att import MultiHeadedAttentionCrossAtt as MultiHeadedAttentionCrossAtt_export -from funasr.modules.positionwise_feed_forward import PositionwiseFeedForwardDecoderSANM +from funasr.models.transformer.positionwise_feed_forward import PositionwiseFeedForwardDecoderSANM from funasr.export.models.modules.feedforward import PositionwiseFeedForwardDecoderSANM as PositionwiseFeedForwardDecoderSANM_export from funasr.export.models.modules.decoder_layer import DecoderLayerSANM as DecoderLayerSANM_export -- Gitblit v1.9.1