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/bin/asr_infer.py | 2 +-
funasr/build_utils/build_model_from_file.py | 2 ++
2 files changed, 3 insertions(+), 1 deletions(-)
diff --git a/funasr/bin/asr_infer.py b/funasr/bin/asr_infer.py
index 806adce..e0e2c09 100644
--- a/funasr/bin/asr_infer.py
+++ b/funasr/bin/asr_infer.py
@@ -83,7 +83,7 @@
# 1. Build ASR model
scorers = {}
asr_model, asr_train_args = build_model_from_file(
- asr_train_config, asr_model_file, cmvn_file, device, mode="asr"
+ asr_train_config, asr_model_file, cmvn_file, device
)
frontend = None
if asr_train_args.frontend is not None and asr_train_args.frontend_conf is not None:
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