aluminumbox commited on
Commit
0933e24
·
verified ·
1 Parent(s): 94b0f9d

Update cosyvoice/cli/frontend.py

Browse files
Files changed (1) hide show
  1. cosyvoice/cli/frontend.py +1 -2
cosyvoice/cli/frontend.py CHANGED
@@ -53,8 +53,7 @@ class CosyVoiceFrontEnd:
53
  option.intra_op_num_threads = 1
54
  self.campplus_session = onnxruntime.InferenceSession(campplus_model, sess_options=option, providers=["CPUExecutionProvider"])
55
  self.speech_tokenizer_session = onnxruntime.InferenceSession(speech_tokenizer_model, sess_options=option,
56
- providers=["CUDAExecutionProvider" if torch.cuda.is_available() else
57
- "CPUExecutionProvider"])
58
  if os.path.exists(spk2info):
59
  self.spk2info = torch.load(spk2info, map_location=self.device)
60
  else:
 
53
  option.intra_op_num_threads = 1
54
  self.campplus_session = onnxruntime.InferenceSession(campplus_model, sess_options=option, providers=["CPUExecutionProvider"])
55
  self.speech_tokenizer_session = onnxruntime.InferenceSession(speech_tokenizer_model, sess_options=option,
56
+ providers=["CPUExecutionProvider"])
 
57
  if os.path.exists(spk2info):
58
  self.spk2info = torch.load(spk2info, map_location=self.device)
59
  else: