| | |
| | | # 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: |
| | |
| | | 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. |
| | |
| | | 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( |