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

Merge pull request #901 from alibaba-damo-academy/dev_lhn

Browse Source 
fix onnx export bug
This commit is contained in:
hnluo 2023-08-29 11:41:51 +08:00 committed by GitHub
commit b22bd68863
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
funasr/export/models/__init__.py
View File

@ -25,8 +25,9 @@ def get_model(model, export_config=None):
elif isinstance(model, BiCifParaformer):
return BiCifParaformer_export(model, **export_config)
elif isinstance(model, ParaformerOnline):
return (ParaformerOnline_encoder_predictor_export(model, model_name="model"),
ParaformerOnline_decoder_export(model, model_name="decoder"))
encoder = ParaformerOnline_encoder_predictor_export(model, model_name="model")
decoder = ParaformerOnline_decoder_export(model, model_name="decoder")
return [encoder, decoder]
elif isinstance(model, Paraformer):
return Paraformer_export(model, **export_config)
elif isinstance(model, Conformer_export):