嘉渊
2023-06-14 5feca0cc1718ca2dea23aecbacf2d2218e13a036
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:
@@ -316,7 +316,7 @@
        # 2. Build Language model
        if lm_train_config is not None:
            lm, lm_train_args = build_model_from_file(
                lm_train_config, lm_file, device
                lm_train_config, lm_file, None, device, task_name="lm"
            )
            scorers["lm"] = lm.lm
@@ -636,7 +636,7 @@
        # 2. Build Language model
        if lm_train_config is not None:
            lm, lm_train_args = build_model_from_file(
                lm_train_config, lm_file, device
                lm_train_config, lm_file, None, device, task_name="lm"
            )
            scorers["lm"] = lm.lm
@@ -1120,7 +1120,7 @@
        # 2. Build Language model
        if lm_train_config is not None:
            lm, lm_train_args = build_model_from_file(
                lm_train_config, lm_file, device
                lm_train_config, lm_file, None, device, task_name="lm"
            )
            lm.to(device)
            scorers["lm"] = lm.lm
@@ -1343,7 +1343,7 @@
        if lm_train_config is not None:
            lm, lm_train_args = build_model_from_file(
                lm_train_config, lm_file, device
                lm_train_config, lm_file, None, device, task_name="lm"
            )
            lm_scorer = lm.lm
        else:
@@ -1659,7 +1659,7 @@
        # 2. Build Language model
        if lm_train_config is not None:
            lm, lm_train_args = build_model_from_file(
                lm_train_config, lm_file, None, device
                lm_train_config, lm_file, None, device, task_name="lm"
            )
            scorers["lm"] = lm.lm