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

export model

Browse Source
This commit is contained in:
游雁 2023-02-10 18:42:52 +08:00
parent 8bf1a6adbb
commit d4b683433a
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
funasr/export/export_model.py
View File

@ -69,7 +69,7 @@ class ASRModelExportParaformer:
model_dir = tag_name model_dir = tag_name
if model_dir.startswith('damo/'): if model_dir.startswith('damo/'):
from modelscope.hub.snapshot_download import snapshot_download from modelscope.hub.snapshot_download import snapshot_download
model_dir = snapshot_download(tag_name, cache_dir=self.cache_dir) model_dir = snapshot_download(model_dir, cache_dir=self.cache_dir)
asr_train_config = os.path.join(model_dir, 'config.yaml') asr_train_config = os.path.join(model_dir, 'config.yaml')
asr_model_file = os.path.join(model_dir, 'model.pb') asr_model_file = os.path.join(model_dir, 'model.pb')
cmvn_file = os.path.join(model_dir, 'am.mvn') cmvn_file = os.path.join(model_dir, 'am.mvn')
@ -87,7 +87,7 @@ class ASRModelExportParaformer:
model, asr_train_args = ASRTask.build_model_from_file( model, asr_train_args = ASRTask.build_model_from_file(
asr_train_config, asr_model_file, cmvn_file, 'cpu' asr_train_config, asr_model_file, cmvn_file, 'cpu'
) )
self.export(model, tag_name) self._export(model, tag_name)
# def export_from_modelscope( # def export_from_modelscope(
# self, # self,