From a8d2941293e189f448b941c249e972f082a46697 Mon Sep 17 00:00:00 2001
From: Xavi Vinaixa <xavi@broomx.com>
Date: Tue, 6 Jun 2023 08:10:51 +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 4e72732..0e51b92 100644
--- a/refacer.py
+++ b/refacer.py
@@ -20,7 +20,7 @@ from insightface.app.common import Face
 from insightface.utils.storage import ensure_available
 
 class RefacerMode(Enum):
-     CPU, CUDA, COREML, TENSORRT = range(1, 4)
+     CPU, CUDA, COREML, TENSORRT = range(1, 5)
 
 class Refacer:
     def __init__(self,force_cpu=False):