From 0379f38dd9aff8b56605272051fb35ef5b13a4ea Mon Sep 17 00:00:00 2001 From: "lyuxiang.lx" Date: Fri, 5 Jul 2024 21:41:50 +0800 Subject: [PATCH] add cpuruntime in provider --- cosyvoice/cli/frontend.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/cosyvoice/cli/frontend.py b/cosyvoice/cli/frontend.py index c1066e8..273679a 100644 --- a/cosyvoice/cli/frontend.py +++ b/cosyvoice/cli/frontend.py @@ -42,7 +42,7 @@ class CosyVoiceFrontEnd: option.graph_optimization_level = onnxruntime.GraphOptimizationLevel.ORT_ENABLE_ALL option.intra_op_num_threads = 1 self.campplus_session = onnxruntime.InferenceSession(campplus_model, sess_options=option, providers=["CPUExecutionProvider"]) - self.speech_tokenizer_session = onnxruntime.InferenceSession(speech_tokenizer_model, sess_options=option, providers=["CUDAExecutionProvider"]) + self.speech_tokenizer_session = onnxruntime.InferenceSession(speech_tokenizer_model, sess_options=option, providers=["CUDAExecutionProvider", "CPUExecutionProvider"]) if os.path.exists(spk2info): self.spk2info = torch.load(spk2info, map_location=self.device) self.instruct = instruct