speech_recognition/FunASR
1
0
Fork 0
mirror of https://github.com/modelscope/FunASR synced 2025-09-15 14:48:36 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

update model name

Browse Source
This commit is contained in:
维石 2024-06-12 17:14:52 +08:00
parent a0c58c8fc6
commit 021035ef4d
3 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
funasr/models/contextual_paraformer/export_meta.py
View File

@ -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

2
funasr/models/seaco_paraformer/export_meta.py
View File

@ -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

2
funasr/utils/export_utils.py
View File

@ -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"))