mirror/stable-diffusion-webui
2
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2024-12-09 07:10:16 +08:00
Code Issues Packages Projects Releases Wiki Activity

even more powerfull fix

Browse Source
This commit is contained in:
Greendayle 2022-10-07 20:46:38 +02:00
parent 54fa613c83
commit fa2ea648db
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
modules/deepbooru.py
View File

@ -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