funasr/models/paraformer/export_meta.py
@@ -31,6 +31,7 @@ model.export_dynamic_axes = types.MethodType(export_dynamic_axes, model) model.export_name = types.MethodType(export_name, model) model.export_name = "model" return model @@ -76,10 +77,11 @@ 0: "batch_size", }, "logits": {0: "batch_size", 1: "logits_length"}, "token_num": {0: "batch_size"} } def export_name( self, ): return "model.onnx" return "model"
