Plachta commited on
Commit
7cfb617
·
verified ·
1 Parent(s): b7e0982

Update modules/cosyvoice_tokenizer/frontend.py

Browse files
modules/cosyvoice_tokenizer/frontend.py CHANGED
@@ -25,7 +25,7 @@ class CosyVoiceFrontEnd:
25
  option = onnxruntime.SessionOptions()
26
  option.graph_optimization_level = onnxruntime.GraphOptimizationLevel.ORT_ENABLE_ALL
27
  option.intra_op_num_threads = 1
28
- self.speech_tokenizer_session = onnxruntime.InferenceSession(speech_tokenizer_model, sess_options=option, providers=["CUDAExecutionProvider" if device == "cuda" and torch.cuda.is_available() else "CPUExecutionProvider"])
29
 
30
  def extract_speech_token(self, speech):
31
  feat = whisper.log_mel_spectrogram(speech, n_mels=128)
 
25
  option = onnxruntime.SessionOptions()
26
  option.graph_optimization_level = onnxruntime.GraphOptimizationLevel.ORT_ENABLE_ALL
27
  option.intra_op_num_threads = 1
28
+ self.speech_tokenizer_session = onnxruntime.InferenceSession(speech_tokenizer_model, sess_options=option, providers=["CPUExecutionProvider"])
29
 
30
  def extract_speech_token(self, speech):
31
  feat = whisper.log_mel_spectrogram(speech, n_mels=128)