From 0f3a4ca84d17d282f5791e9eee383948d67dfe1b Mon Sep 17 00:00:00 2001
From: 嘉渊 <wangjiaming.wjm@alibaba-inc.com>
Date: 星期三, 14 六月 2023 17:29:57 +0800
Subject: [PATCH] update repo
---
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 51de5b0..5488c10 100644
--- a/funasr/build_utils/build_model_from_file.py
+++ b/funasr/build_utils/build_model_from_file.py
@@ -17,6 +17,7 @@
model_file: Union[Path, str] = None,
cmvn_file: Union[Path, str] = None,
device: str = "cpu",
+ task_name: str = "asr",
mode: str = "paraformer",
):
"""Build model from the files.
@@ -44,6 +45,7 @@
if cmvn_file is not None:
args["cmvn_file"] = cmvn_file
args = argparse.Namespace(**args)
+ args.task_name = task_name
model = build_model(args)
if not isinstance(model, FunASRModel):
raise RuntimeError(
--
Gitblit v1.9.1