From 874ed3aaac591e2d1c29087b5f4e86ba66272f8b Mon Sep 17 00:00:00 2001
From: 维石 <shixian.shi@alibaba-inc.com>
Date: 星期五, 26 七月 2024 11:40:23 +0800
Subject: [PATCH] bug fix for onnx infer
---
funasr/models/seaco_paraformer/export_meta.py | 9 ++++-----
1 files changed, 4 insertions(+), 5 deletions(-)
diff --git a/funasr/models/seaco_paraformer/export_meta.py b/funasr/models/seaco_paraformer/export_meta.py
index a246a29..947f4d9 100644
--- a/funasr/models/seaco_paraformer/export_meta.py
+++ b/funasr/models/seaco_paraformer/export_meta.py
@@ -53,7 +53,7 @@
0: "num_hotwords",
},
"hw_embed": {
- 0: "num_hotwords",
+ 1: "num_hotwords",
},
}
@@ -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"
--
Gitblit v1.9.1