speech_recognition/FunASR
1
0
Fork 0
mirror of https://github.com/modelscope/FunASR synced 2025-09-15 14:48:36 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

dcos

Browse Source
This commit is contained in:
游雁 2024-07-30 17:45:13 +08:00
parent 09f0f20545
commit d238a5ab44
2 changed files with 8 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
funasr/auto/auto_model.py
View File

@ -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

9
funasr/models/mossformer/mossformer_encoder.py
View File

@ -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):