FUNASR训练
parent: 3999789c | 补丁 | 提交 | ignore whitespace
游雁
2023-04-20 3845722f8fecaa568e052aea610961b9271ddd5a 
Merge branch 'main' of github.com:alibaba-damo-academy/FunASR
add
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