From 7e628a3b83f5ffd073d6e4271f5f4ec16c190282 Mon Sep 17 00:00:00 2001
From: Xavi Vinaixa <xavi@broomx.com>
Date: Tue, 6 Jun 2023 08:04:33 +0200
Subject: [PATCH] Incredible speed boost with TensorrtExecutionProvider #5.

---
 refacer.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/refacer.py b/refacer.py
index df855f1..c52234d 100644
--- a/refacer.py
+++ b/refacer.py
@@ -49,7 +49,7 @@ class Refacer:
             self.use_num_cpus = mp.cpu_count()-1
             print(f"CoreML mode with providers {self.providers}")
             self.sess_options.intra_op_num_threads = int(self.use_num_cpus/2)
-        elif 'CUDAExecutionProvider' in self.providers:
+        elif 'CUDAExecutionProvider' in self.providers or 'TensorrtExecutionProvider' in self.providers:
             self.mode = RefacerMode.CUDA
             self.use_num_cpus = 1
             self.sess_options.intra_op_num_threads = 1