speech_recognition/FunASR
1
0
Fork 0
mirror of https://github.com/modelscope/FunASR synced 2025-09-15 14:48:36 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

update repo

Browse Source
This commit is contained in:
嘉渊 2023-06-14 17:29:57 +08:00
parent 9dc9482300
commit 0f3a4ca84d
2 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
funasr/bin/asr_infer.py
View File

@ -83,7 +83,7 @@ class Speech2Text:
# 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:

2
funasr/build_utils/build_model_from_file.py
View File

@ -17,6 +17,7 @@ def build_model_from_file(
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 @@ def build_model_from_file(
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(