From 1f0c3b4bb5288acc20c31b37ed070d691e94b3f8 Mon Sep 17 00:00:00 2001
From: henryruhs <info@henryruhs.com>
Date: Sat, 3 Jun 2023 13:37:15 +0200
Subject: [PATCH] Double GPU performance

---
 roop/core.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/roop/core.py b/roop/core.py
index de5d7f6..50ad051 100644
--- a/roop/core.py
+++ b/roop/core.py
@@ -1,12 +1,14 @@
 #!/usr/bin/env python3
 
+import os
+# reducing num threads doubles performance of gpu-mode
+os.environ['OMP_NUM_THREADS'] = '1'
 import platform
 import signal
 import sys
 import shutil
 import glob
 import argparse
-import os
 import torch
 from pathlib import Path
 import tkinter as tk
@@ -16,7 +18,6 @@ from tkinter.filedialog import asksaveasfilename
 import webbrowser
 import cv2
 import threading
-from threadpoolctl import threadpool_limits
 from PIL import Image, ImageTk
 
 import roop.globals