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

update export name

Browse Source
This commit is contained in:
维石 2024-05-30 14:22:32 +08:00
parent 64d7032073
commit 23eed38e9f
2 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
funasr/models/bicif_paraformer/export_meta.py
View File

@ -29,7 +29,8 @@ def export_rebuild_model(model, **kwargs):
model.export_input_names = types.MethodType(export_input_names, model)
model.export_output_names = types.MethodType(export_output_names, model)
model.export_dynamic_axes = types.MethodType(export_dynamic_axes, model)
model.export_name = types.MethodType(export_name, model)
model.export_name = "model"
return model

4
funasr/models/contextual_paraformer/export_meta.py
View File

@ -59,7 +59,9 @@ def export_rebuild_model(model, **kwargs):
backbone_model.export_dynamic_axes = types.MethodType(
export_backbone_dynamic_axes, backbone_model
)
backbone_model.export_name = types.MethodType(export_backbone_name, backbone_model)
embedder_model.export_name = "model_eb"
backbone_model.export_name = "model_bb"
return backbone_model, embedder_model