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 spk inference

Browse Source
This commit is contained in:
shixian.shi 2023-12-06 11:29:43 +08:00
parent b3fcd42bf6
commit e54535e5eb
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
funasr/bin/asr_inference_launch.py
View File

@ -952,10 +952,13 @@ def inference_paraformer_vad_speaker(
##### speaker_verification #####
##################################
# load sv model
sv_model_dict = torch.load(sv_model_file)
sv_model = getattr(sv_module, sv_model_config['sv_model_class'])(**sv_model_config['models_config'])
if ngpu > 0:
sv_model_dict = torch.load(sv_model_file)
sv_model = getattr(sv_module, sv_model_config['sv_model_class'])(**sv_model_config['models_config'])
sv_model.cuda()
else:
sv_model_dict = torch.load(sv_model_file, map_location=torch.device('cpu'))
sv_model = getattr(sv_module, sv_model_config['sv_model_class'])(**sv_model_config['models_config'])
sv_model.load_state_dict(sv_model_dict)
print(f'load sv model params: {sv_model_file}')
sv_model.eval()