diff --git a/funasr/models/fun_asr_nano/model.py b/funasr/models/fun_asr_nano/model.py index 2e11a39b3..49294d405 100644 --- a/funasr/models/fun_asr_nano/model.py +++ b/funasr/models/fun_asr_nano/model.py @@ -9,7 +9,6 @@ import torch import torch.nn as nn -from funasr import AutoModel from funasr.metrics.compute_acc import compute_accuracy from funasr.register import tables from funasr.train_utils.device_funcs import force_gatherable, to_device @@ -42,6 +41,7 @@ def __init__( "activation_checkpoint", False ) if hub == "ms": + from funasr import AutoModel model = AutoModel(model=audio_encoder, model_revision="master") audio_encoder_output_size = ( model.model.encoder_output_size