update model name

维石

2024-06-12 021035ef4deeb715d0eea01a9f00d0f2404e06ed

update model name

 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


 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


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