From d238a5ab4450aad636bbbc60d67335ca59b3bd9c Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E6=B8=B8=E9=9B=81?= Date: Tue, 30 Jul 2024 17:45:13 +0800 Subject: [PATCH] dcos --- funasr/auto/auto_model.py | 3 +++ funasr/models/mossformer/mossformer_encoder.py | 9 +++++---- 2 files changed, 8 insertions(+), 4 deletions(-) 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):