From 27e8ba254adbe9f0df50b4969c5d2e66e0c7c0de Mon Sep 17 00:00:00 2001
From: jaluik <37829041+jaluik@users.noreply.github.com>
Date: 星期二, 09 一月 2024 10:57:03 +0800
Subject: [PATCH] docs: update docker tag (#1227)
---
funasr/bin/asr_inference_launch.py | 7 +++++--
1 files changed, 5 insertions(+), 2 deletions(-)
diff --git a/funasr/bin/asr_inference_launch.py b/funasr/bin/asr_inference_launch.py
index 59e61ee..f34bfb2 100644
--- a/funasr/bin/asr_inference_launch.py
+++ b/funasr/bin/asr_inference_launch.py
@@ -952,10 +952,13 @@
##### 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()
--
Gitblit v1.9.1