Spaces:

TeraTTS
/

TTS

Running

Den4ikAI commited on Sep 22, 2023

Commit

f18df21

1 Parent(s): dfe40db

Update infer_onnx.py

Files changed (1) hide show

infer_onnx.py CHANGED Viewed

@@ -18,8 +18,7 @@ class TTS:
                               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")):
             from tokenizer import TokenizerG2P

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