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 repo

Browse Source
This commit is contained in:
嘉渊 2023-06-19 13:45:11 +08:00
parent a17c5cc3e7
commit e2f8fa4b64
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
funasr/build_utils/build_model_from_file.py
View File

@ -74,7 +74,10 @@ def build_model_from_file(
model_dict = torch.load(model_file, map_location=device)
if task_name == "diar" and mode == "sond":
model_dict = fileter_model_dict(model_dict, model.state_dict())
model.load_state_dict(model_dict)
if task_name == "vad":
model.encoder.load_state_dict(model_dict)
else:
model.load_state_dict(model_dict)
if model_name_pth is not None and not os.path.exists(model_name_pth):
torch.save(model_dict, model_name_pth)
logging.info("model_file is saved to pth: {}".format(model_name_pth))