diff options
| author | Greendayle <Greendayle> | 2022-10-07 18:58:30 +0000 |
|---|---|---|
| committer | Greendayle <Greendayle> | 2022-10-07 18:58:30 +0000 |
| commit | 5f12e7efd92ad802742f96788b4be3249ad02829 (patch) | |
| tree | a47f99457cd4ee21468e317256a7f56930b3b978 /modules/deepbooru.py | |
| parent | fa2ea648db81f5723bb5d722f2fe0ebd7dfc319a (diff) | |
| download | stable-diffusion-webui-gfx803-5f12e7efd92ad802742f96788b4be3249ad02829.tar.gz stable-diffusion-webui-gfx803-5f12e7efd92ad802742f96788b4be3249ad02829.tar.bz2 stable-diffusion-webui-gfx803-5f12e7efd92ad802742f96788b4be3249ad02829.zip | |
linux test
Diffstat (limited to 'modules/deepbooru.py')
| -rw-r--r-- | modules/deepbooru.py | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/modules/deepbooru.py b/modules/deepbooru.py index 60094336..781b2249 100644 --- a/modules/deepbooru.py +++ b/modules/deepbooru.py @@ -1,6 +1,6 @@ import os.path from concurrent.futures import ProcessPoolExecutor - +from multiprocessing import get_context def _load_tf_and_return_tags(pil_image, threshold): @@ -66,7 +66,8 @@ def subprocess_init_no_cuda(): def get_deepbooru_tags(pil_image, threshold=0.5): - with ProcessPoolExecutor(initializer=subprocess_init_no_cuda) as executor: + context = get_context('spawn') + with ProcessPoolExecutor(initializer=subprocess_init_no_cuda, mp_context=context) 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 |