维石
2024-05-28 e7351db81b3bfc4000633eca274c46893d68f64e
funasr/models/seaco_paraformer/export_meta.py
@@ -109,7 +109,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
@@ -192,6 +194,3 @@
        "pre_acoustic_embeds": {1: "feats_length1"},
    }
def export_backbone_name(self):
    return "model.onnx"