From c644ac8f58895b9e29e9cfca79465fd2c0efaa5a Mon Sep 17 00:00:00 2001 From: 游雁 <zhifu.gzf@alibaba-inc.com> Date: 星期二, 21 十一月 2023 14:09:01 +0800 Subject: [PATCH] funasr v2 setup --- funasr/models/encoder/mossformer_encoder.py | 6 ++++-- 1 files changed, 4 insertions(+), 2 deletions(-) diff --git a/funasr/models/encoder/mossformer_encoder.py b/funasr/models/encoder/mossformer_encoder.py index 54d80ca..f7d9c47 100644 --- a/funasr/models/encoder/mossformer_encoder.py +++ b/funasr/models/encoder/mossformer_encoder.py @@ -1,8 +1,10 @@ import torch import torch.nn as nn import torch.nn.functional as F - -from rotary_embedding_torch import RotaryEmbedding +try: + from rotary_embedding_torch import RotaryEmbedding +except: + raise "Please install rotary_embedding_torch by: \n pip install -U funasr[all]" from funasr.modules.layer_norm import GlobalLayerNorm, CumulativeLayerNorm, ScaleNorm from funasr.modules.embedding import ScaledSinuEmbedding from funasr.modules.mossformer import FLASH_ShareA_FFConvM -- Gitblit v1.9.1