ru

Running

Den4ikAI commited on Sep 22, 2023

Commit

dfe40db

•

1 Parent(s): aae6676

Update infer_onnx.py

Files changed (1) hide show

infer_onnx.py CHANGED Viewed

@@ -18,7 +18,7 @@ class TTS:
                               local_dir_use_symlinks=False
                             )
-        sess_options = onnxruntime.SessionOptions()
         self.model = onnxruntime.InferenceSession(os.path.join(model_dir, "exported/model.onnx"), sess_options=sess_options)
         if os.path.exists(os.path.join(model_dir, "exported/dictionary.txt")):

                               local_dir_use_symlinks=False
                             )
+        sess_options = ['CPUExecutionProvider']
         self.model = onnxruntime.InferenceSession(os.path.join(model_dir, "exported/model.onnx"), sess_options=sess_options)
         if os.path.exists(os.path.join(model_dir, "exported/dictionary.txt")):