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/encoder/sanm_encoder.py | 6 +++--- 1 files changed, 3 insertions(+), 3 deletions(-) diff --git a/funasr/export/models/encoder/sanm_encoder.py b/funasr/export/models/encoder/sanm_encoder.py index d1b4b1e..7ef863e 100644 --- a/funasr/export/models/encoder/sanm_encoder.py +++ b/funasr/export/models/encoder/sanm_encoder.py @@ -3,12 +3,12 @@ from funasr.export.utils.torch_function import MakePadMask from funasr.export.utils.torch_function import sequence_mask -from funasr.modules.attention import MultiHeadedAttentionSANM +from funasr.models.transformer.attention import MultiHeadedAttentionSANM from funasr.export.models.modules.multihead_att import MultiHeadedAttentionSANM as MultiHeadedAttentionSANM_export from funasr.export.models.modules.encoder_layer import EncoderLayerSANM as EncoderLayerSANM_export -from funasr.modules.positionwise_feed_forward import PositionwiseFeedForward +from funasr.models.transformer.positionwise_feed_forward import PositionwiseFeedForward from funasr.export.models.modules.feedforward import PositionwiseFeedForward as PositionwiseFeedForward_export -from funasr.modules.embedding import StreamSinusoidalPositionEncoder +from funasr.models.transformer.embedding import StreamSinusoidalPositionEncoder class SANMEncoder(nn.Module): -- Gitblit v1.9.1