From 7e9696f15635a83ff810cc86b79c0932cb4c8a27 Mon Sep 17 00:00:00 2001
From: Djraemon <154407769+Djraemon@users.noreply.github.com>
Date: 星期一, 04 十一月 2024 11:04:52 +0800
Subject: [PATCH] Fix audio format 2.0 (#2186)
---
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