维石
2024-05-30 23eed38e9fbe547bab63a99a560c0ad33a5f13b8
update export name
2个文件已修改
7 ■■■■ 已修改文件
funasr/models/bicif_paraformer/export_meta.py 3 ●●●● 补丁 | 查看 | 原始文档 | blame | 历史
funasr/models/contextual_paraformer/export_meta.py 4 ●●● 补丁 | 查看 | 原始文档 | blame | 历史
funasr/models/bicif_paraformer/export_meta.py
@@ -29,7 +29,8 @@
    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
funasr/models/contextual_paraformer/export_meta.py
@@ -59,7 +59,9 @@
    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