From 5c9bdfa238560b9c4a8d8a7d4b56c28496a3e094 Mon Sep 17 00:00:00 2001
From: aky15 <ankeyuthu@gmail.com>
Date: 星期四, 14 九月 2023 16:12:53 +0800
Subject: [PATCH] Merge pull request #952 from alibaba-damo-academy/aky15-patch-1
---
funasr/build_utils/build_model_from_file.py | 2 ++
1 files changed, 2 insertions(+), 0 deletions(-)
diff --git a/funasr/build_utils/build_model_from_file.py b/funasr/build_utils/build_model_from_file.py
index 26542cd..65e0d5f 100644
--- a/funasr/build_utils/build_model_from_file.py
+++ b/funasr/build_utils/build_model_from_file.py
@@ -70,6 +70,8 @@
model.load_state_dict(model_dict)
else:
model_dict = torch.load(model_file, map_location=device)
+ if task_name == "ss":
+ model_dict = model_dict['model']
if task_name == "diar" and mode == "sond":
model_dict = fileter_model_dict(model_dict, model.state_dict())
if task_name == "vad":
--
Gitblit v1.9.1