From d9884e41fefd7867afde6b0b1233baddc038e72f Mon Sep 17 00:00:00 2001
From: Xavi Vinaixa <xavi@broomx.com>
Date: Tue, 6 Jun 2023 12:26:11 +0200
Subject: [PATCH] Disabled TensorrtExecutionProvider due to a bug, it will now
 run with CUDAExecutionProvider on NVIDIA until we resolve error #9 #10

---
 refacer.py | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

diff --git a/refacer.py b/refacer.py
index e302886..1f5d161 100644
--- a/refacer.py
+++ b/refacer.py
@@ -52,6 +52,11 @@ class Refacer:
             self.use_num_cpus = mp.cpu_count()-1
             self.sess_options.intra_op_num_threads = int(self.use_num_cpus/3)
             print(f"CoreML mode with providers {self.providers}")
+        elif 'CUDAExecutionProvider' in self.providers:
+            self.mode = RefacerMode.CUDA
+            self.use_num_cpus = 1
+            self.sess_options.intra_op_num_threads = 1
+            print(f"CUDA mode with providers {self.providers}")
         elif 'TensorrtExecutionProvider' in self.providers:
             self.mode = RefacerMode.TENSORRT
             #self.use_num_cpus = 1
@@ -59,11 +64,6 @@ class Refacer:
             self.use_num_cpus = mp.cpu_count()-1
             self.sess_options.intra_op_num_threads = int(self.use_num_cpus/3)
             print(f"TENSORRT mode with providers {self.providers}")
-        elif 'CUDAExecutionProvider' in self.providers:
-            self.mode = RefacerMode.CUDA
-            self.use_num_cpus = 1
-            self.sess_options.intra_op_num_threads = 1
-            print(f"CUDA mode with providers {self.providers}")
         
 
     def __init_apps(self):