FunASR

FunASR copied to clipboard

🐛 Bug

To Reproduce

Steps to reproduce the behavior (always include the command you ran):

1. Run cmd '....' (fun) it@ai01:/opt/ai_users/abdurakhim/paraformer_train$ /home/it/anaconda3/envs/fun/bin/python /opt/ai_users/abdurakhim/paraformer_train/application/export_model.py funasr version: 1.2.6. Check update of funasr, and it would cost few times. You may disable it by set disable_update=True in AutoModel You are using the latest version of funasr-1.2.6 WARNING:root:Using legacy_rel_pos and it will be deprecated in the future. WARNING:root:Using legacy_rel_selfattn and it will be deprecated in the future. rtf_avg: 0.030: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 1/1 [00:00<00:00, 7.01it/s] [{'key': 'test', 'text': 't õ XTABBURILIBQARADI'}] Traceback (most recent call last): File "/opt/ai_users/abdurakhim/paraformer_train/application/export_model.py", line 16, in res = model.export(type="onnx", quantize=False, opset_version=13, device='cpu') ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/home/it/anaconda3/envs/fun/lib/python3.11/site-packages/funasr/auto/auto_model.py", line 670, in export export_dir = export_utils.export(model=model, data_in=data_list, **kwargs) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/home/it/anaconda3/envs/fun/lib/python3.11/site-packages/funasr/utils/export_utils.py", line 13, in export model_scripts = model.export(**kwargs) ^^^^^^^^^^^^^^^^^^^^^^ File "/home/it/anaconda3/envs/fun/lib/python3.11/site-packages/funasr/models/paraformer/model.py", line 603, in export models = export_rebuild_model(model=self, **kwargs) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/home/it/anaconda3/envs/fun/lib/python3.11/site-packages/funasr/models/paraformer/export_meta.py", line 15, in export_rebuild_model model.encoder = encoder_class(model.encoder, onnx=is_onnx) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ TypeError: 'NoneType' object is not callable

Code sample

from funasr import AutoModel

model = AutoModel(
    model="/opt/ai_users/abdurakhim/paraformer_train/checkpoints/paraformer_uzbek_natural_v1",
    model_path="/opt/ai_users/abdurakhim/paraformer_train/checkpoints/paraformer_uzbek_natural_v1/model.pt.best", 
    config_path="/opt/ai_users/abdurakhim/paraformer_train/checkpoints/paraformer_uzbek_natural_v1", 
    config_name="config.yaml"
)

result = model.generate(
            input="/opt/ai_users/abdurakhim/paraformer_train/test.wav",
            batch_size_s=300
        )
print(result)
res = model.export(type="onnx", quantize=False, opset_version=13, device='cpu')

Environment

• OS (e.g., Linux):
• FunASR Version (e.g., 1.0.0): 1.2.6
• ModelScope Version (e.g., 1.11.0):
• PyTorch Version (e.g., 2.0.0): 2.6.0
• How you installed funasr (pip, source): pip
• Python version: 3.11

Additional context

Anyone know what's happening there? I used default config from repository.