Incredible speed boost with TensorrtExecutionProvider #5.

2023-06-06 08:10:51 +02:00
parent 4a24df4755
commit a8d2941293
1 changed files with 1 additions and 1 deletions
--- 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):