Update infer_onnx.py
Browse files- infer_onnx.py +1 -2
infer_onnx.py
CHANGED
@@ -18,8 +18,7 @@ class TTS:
|
|
18 |
local_dir_use_symlinks=False
|
19 |
)
|
20 |
|
21 |
-
|
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
|