From 23eed38e9fbe547bab63a99a560c0ad33a5f13b8 Mon Sep 17 00:00:00 2001
From: 维石 <shixian.shi@alibaba-inc.com>
Date: 星期四, 30 五月 2024 14:22:32 +0800
Subject: [PATCH] update export name
---
funasr/models/bicif_paraformer/export_meta.py | 3 ++-
funasr/models/contextual_paraformer/export_meta.py | 4 +++-
2 files changed, 5 insertions(+), 2 deletions(-)
diff --git a/funasr/models/bicif_paraformer/export_meta.py b/funasr/models/bicif_paraformer/export_meta.py
index e9d0a25..75171f4 100644
--- a/funasr/models/bicif_paraformer/export_meta.py
+++ b/funasr/models/bicif_paraformer/export_meta.py
@@ -29,7 +29,8 @@
model.export_input_names = types.MethodType(export_input_names, model)
model.export_output_names = types.MethodType(export_output_names, model)
model.export_dynamic_axes = types.MethodType(export_dynamic_axes, model)
- model.export_name = types.MethodType(export_name, model)
+
+ model.export_name = "model"
return model
diff --git a/funasr/models/contextual_paraformer/export_meta.py b/funasr/models/contextual_paraformer/export_meta.py
index 602057f..7543789 100644
--- a/funasr/models/contextual_paraformer/export_meta.py
+++ b/funasr/models/contextual_paraformer/export_meta.py
@@ -59,7 +59,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_bb"
return backbone_model, embedder_model
--
Gitblit v1.9.1