From 7fbb56719b4c9b0f85e1ca095a2ae2fe401a7b2d Mon Sep 17 00:00:00 2001
From: 雾聪 <wucong.lyb@alibaba-inc.com>
Date: 星期一, 26 六月 2023 16:52:44 +0800
Subject: [PATCH] add check for local models

---
 funasr/bin/sv_infer.py |    3 ++-
 1 files changed, 2 insertions(+), 1 deletions(-)

diff --git a/funasr/bin/sv_infer.py b/funasr/bin/sv_infer.py
index fd0d666..6e861da 100755
--- a/funasr/bin/sv_infer.py
+++ b/funasr/bin/sv_infer.py
@@ -50,7 +50,8 @@
             model_file=sv_model_file,
             cmvn_file=None,
             device=device,
-            task_name="sv"
+            task_name="sv",
+            mode="sv",
         )
         logging.info("sv_model: {}".format(sv_model))
         logging.info("model parameter number: {}".format(statistic_model_parameters(sv_model)))

--
Gitblit v1.9.1