Shi Xian
2024-06-18 6c467e6f0abfc6d20d0621fbbf67b4dbd81776cc
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"
    return backbone_model, embedder_model
@@ -198,6 +200,3 @@
        "us_cif_peak": {0: "batch_size", 1: "alphas_length"},
    }
def export_backbone_name(self):
    return "model.onnx"