FUNASR训练
parent: 133a1158 | 补丁 | 提交 | ignore whitespace
shixian.shi
2023-04-20 1f6370e350b16dd18499ce2ae7f32ad9216a62f8 
bug fix
1个文件已修改
2 ■■■■■ 已修改文件
funasr/bin/tp_inference.py 2 ●●●●● 补丁 | 查看 | 原始文档 | blame | 历史 
 funasr/bin/tp_inference.py


@@ -54,7 +54,7 @@


        assert check_argument_types()


        # 1. Build ASR model


        tp_model, tp_train_args = ASRTask.build_model_from_file(


            timestamp_infer_config, timestamp_model_file, device


            timestamp_infer_config, timestamp_model_file, device=device


        )


        if 'cuda' in device:


            tp_model = tp_model.cuda()  # force model to cuda