export

funasr/models/target_delay_transformer.py

@@ -7,7 +7,7 @@ import torch.nn as nn
 
 from funasr.modules.embedding import SinusoidalPositionEncoder
 #from funasr.models.encoder.transformer_encoder import TransformerEncoder as Encoder
-from funasr.punctuation.sanm_encoder import SANMEncoder as Encoder
+from funasr.models.encoder.sanm_encoder import SANMEncoder as Encoder
 #from funasr.modules.mask import subsequent_n_mask
 from funasr.train.abs_model import AbsPunctuation
 

funasr/models/vad_realtime_transformer.py

@@ -6,7 +6,7 @@ import torch
 import torch.nn as nn
 
 from funasr.modules.embedding import SinusoidalPositionEncoder
-from funasr.punctuation.sanm_encoder import SANMVadEncoder as Encoder
+from funasr.models.encoder.sanm_encoder import SANMVadEncoder as Encoder
 from funasr.train.abs_model import AbsPunctuation