From f2e4dd93f448dff8122686f2c39c5630d81a7c39 Mon Sep 17 00:00:00 2001
From: Xavi Vinaixa <xavi@broomx.com>
Date: Tue, 6 Jun 2023 12:39:20 +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 | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/refacer.py b/refacer.py
index 6307237..ed310f6 100644
--- a/refacer.py
+++ b/refacer.py
@@ -56,6 +56,8 @@ class Refacer:
             self.mode = RefacerMode.CUDA
             self.use_num_cpus = 1
             self.sess_options.intra_op_num_threads = 1
+            if 'TensorrtExecutionProvider' in self.providers:
+                self.providers.remove('TensorrtExecutionProvider')
             print(f"CUDA mode with providers {self.providers}")
         """
         elif 'TensorrtExecutionProvider' in self.providers: