From fa2ea648db81f5723bb5d722f2fe0ebd7dfc319a Mon Sep 17 00:00:00 2001
From: Greendayle <Greendayle>
Date: Fri, 7 Oct 2022 20:46:38 +0200
Subject: [PATCH] even more powerfull fix

---
 modules/deepbooru.py | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/modules/deepbooru.py b/modules/deepbooru.py
index 79dc59bd..60094336 100644
--- a/modules/deepbooru.py
+++ b/modules/deepbooru.py
@@ -60,8 +60,13 @@ def _load_tf_and_return_tags(pil_image, threshold):
     return ', '.join(result_tags_out).replace('_', ' ').replace(':', ' ')
 
 
+def subprocess_init_no_cuda():
+    import os
+    os.environ["CUDA_VISIBLE_DEVICES"] = "-1"
+
+
 def get_deepbooru_tags(pil_image, threshold=0.5):
-    with ProcessPoolExecutor() as executor:
-        f = executor.submit(_load_tf_and_return_tags, pil_image, threshold)
+    with ProcessPoolExecutor(initializer=subprocess_init_no_cuda) as executor:
+        f = executor.submit(_load_tf_and_return_tags, pil_image, threshold, )
         ret = f.result()  # will rethrow any exceptions
     return ret
\ No newline at end of file