FUNASR训练
parent: 133a1158 | 补丁 | 提交 | ignore whitespace
Lizerui9926
2023-04-20 aa54e65287ff99ab364bc9893bff79900c8c4cc8 
Merge pull request #386 from alibaba-damo-academy/dev_sx

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