From 021035ef4deeb715d0eea01a9f00d0f2404e06ed Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E7=BB=B4=E7=9F=B3?= Date: Wed, 12 Jun 2024 17:14:52 +0800 Subject: [PATCH] update model name --- funasr/models/contextual_paraformer/export_meta.py | 2 +- funasr/models/seaco_paraformer/export_meta.py | 2 +- funasr/utils/export_utils.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 5fce7acd5..9d3a63b8a 100644 --- a/funasr/models/contextual_paraformer/export_meta.py +++ b/funasr/models/contextual_paraformer/export_meta.py @@ -76,7 +76,7 @@ def export_rebuild_model(model, **kwargs): ) 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 db27c914a..94018a594 100644 --- a/funasr/models/seaco_paraformer/export_meta.py +++ b/funasr/models/seaco_paraformer/export_meta.py @@ -111,7 +111,7 @@ def export_rebuild_model(model, **kwargs): ) 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 32a88abd9..5a98847b8 100644 --- a/funasr/utils/export_utils.py +++ b/funasr/utils/export_utils.py @@ -196,4 +196,4 @@ def _bladedisc_opt_for_encdec(model, path, enable_fp16): 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"))