Den4ikAI commited on
Commit
f18df21
1 Parent(s): dfe40db

Update infer_onnx.py

Browse files
Files changed (1) hide show
  1. infer_onnx.py +1 -2
infer_onnx.py CHANGED
@@ -18,8 +18,7 @@ class TTS:
18
  local_dir_use_symlinks=False
19
  )
20
 
21
- sess_options = ['CPUExecutionProvider']
22
- self.model = onnxruntime.InferenceSession(os.path.join(model_dir, "exported/model.onnx"), sess_options=sess_options)
23
 
24
  if os.path.exists(os.path.join(model_dir, "exported/dictionary.txt")):
25
  from tokenizer import TokenizerG2P
 
18
  local_dir_use_symlinks=False
19
  )
20
 
21
+ self.model = onnxruntime.InferenceSession(os.path.join(model_dir, "exported/model.onnx"), providers=['CPUExecutionProvider'])
 
22
 
23
  if os.path.exists(os.path.join(model_dir, "exported/dictionary.txt")):
24
  from tokenizer import TokenizerG2P