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,6 +77,7 @@ 0: "batch_size", }, "logits": {0: "batch_size", 1: "logits_length"}, "token_num": {0: "batch_size"} }