From 021035ef4deeb715d0eea01a9f00d0f2404e06ed Mon Sep 17 00:00:00 2001
From: 维石 <shixian.shi@alibaba-inc.com>
Date: 星期三, 12 六月 2024 17:14:52 +0800
Subject: [PATCH] update model name
---
funasr/utils/export_utils.py | 2 +-
funasr/models/seaco_paraformer/export_meta.py | 2 +-
funasr/models/contextual_paraformer/export_meta.py | 2 +-
3 files changed, 3 insertions(+), 3 deletions(-)
diff --git a/funasr/models/contextual_paraformer/export_meta.py b/funasr/models/contextual_paraformer/export_meta.py
index 5fce7ac..9d3a63b 100644
--- a/funasr/models/contextual_paraformer/export_meta.py
+++ b/funasr/models/contextual_paraformer/export_meta.py
@@ -76,7 +76,7 @@
)
embedder_model.export_name = "model_eb"
- backbone_model.export_name = "model_bb"
+ backbone_model.export_name = "model"
return backbone_model, embedder_model
diff --git a/funasr/models/seaco_paraformer/export_meta.py b/funasr/models/seaco_paraformer/export_meta.py
index db27c91..94018a5 100644
--- a/funasr/models/seaco_paraformer/export_meta.py
+++ b/funasr/models/seaco_paraformer/export_meta.py
@@ -111,7 +111,7 @@
)
embedder_model.export_name = "model_eb"
- backbone_model.export_name = "model_bb"
+ backbone_model.export_name = "model"
return backbone_model, embedder_model
diff --git a/funasr/utils/export_utils.py b/funasr/utils/export_utils.py
index 32a88ab..5a98847 100644
--- a/funasr/utils/export_utils.py
+++ b/funasr/utils/export_utils.py
@@ -196,4 +196,4 @@
model.encoder = _bladedisc_opt(model.encoder, input_data[:2])
model.decoder = _bladedisc_opt(model.decoder, tuple(decoder_inputs))
model_script = torch.jit.trace(model, input_data)
- model_script.save(os.path.join(path, f"{model.export_name}.torchscripts"))
+ model_script.save(os.path.join(path, f"{model.export_name}_blade.torchscripts"))
--
Gitblit v1.9.1