嘉渊 2 سال پیش
والد
کامیت
0f3a4ca84d
2فایلهای تغییر یافته به همراه3 افزوده شده و 1 حذف شده
  1. 1 1
      funasr/bin/asr_infer.py
  2. 2 0
      funasr/build_utils/build_model_from_file.py

+ 1 - 1
funasr/bin/asr_infer.py

@@ -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 - 0
funasr/build_utils/build_model_from_file.py

@@ -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(