diff --git a/funasr/auto/auto_model.py b/funasr/auto/auto_model.py index ca1f20263..f735f1865 100644 --- a/funasr/auto/auto_model.py +++ b/funasr/auto/auto_model.py @@ -114,6 +114,9 @@ class AutoModel: try: from funasr.utils.version_checker import check_for_update + print( + "Check update of funasr, and it would cost few times. You may disable it by set `disable_update=True` in AutoModel" + ) check_for_update(disable=kwargs.get("disable_update", False)) except: pass diff --git a/funasr/models/mossformer/mossformer_encoder.py b/funasr/models/mossformer/mossformer_encoder.py index 888aef831..a691d8c62 100644 --- a/funasr/models/mossformer/mossformer_encoder.py +++ b/funasr/models/mossformer/mossformer_encoder.py @@ -5,12 +5,13 @@ import torch.nn.functional as F try: from rotary_embedding_torch import RotaryEmbedding except: - print( - "If you want use mossformer, lease install rotary_embedding_torch by: \n pip install -U rotary_embedding_torch" - ) + # print( + # "If you want use mossformer, lease install rotary_embedding_torch by: \n pip install -U rotary_embedding_torch" + # ) + pass from funasr.models.transformer.layer_norm import GlobalLayerNorm, CumulativeLayerNorm, ScaleNorm from funasr.models.transformer.embedding import ScaledSinuEmbedding -from funasr.models.transformer.mossformer import FLASH_ShareA_FFConvM +from funasr.models.mossformer.mossformer import FLASH_ShareA_FFConvM def select_norm(norm, dim, shape):