diff options
| author | Greendayle <Greendayle> | 2022-10-07 18:37:43 +0000 |
|---|---|---|
| committer | Greendayle <Greendayle> | 2022-10-07 18:37:43 +0000 |
| commit | 54fa613c8391e3973cca9d94cdf539061932508b (patch) | |
| tree | 8b43cdf6510145e4754008b00b5eb5be0133d895 /modules/deepbooru.py | |
| parent | 537da7a304adff95fb2ed8337f7a764d08f67c46 (diff) | |
| download | stable-diffusion-webui-gfx803-54fa613c8391e3973cca9d94cdf539061932508b.tar.gz stable-diffusion-webui-gfx803-54fa613c8391e3973cca9d94cdf539061932508b.tar.bz2 stable-diffusion-webui-gfx803-54fa613c8391e3973cca9d94cdf539061932508b.zip | |
loading tf only in interrogation process
Diffstat (limited to 'modules/deepbooru.py')
| -rw-r--r-- | modules/deepbooru.py | 7 |
1 files changed, 4 insertions, 3 deletions
diff --git a/modules/deepbooru.py b/modules/deepbooru.py index fb5018a6..79dc59bd 100644 --- a/modules/deepbooru.py +++ b/modules/deepbooru.py @@ -1,12 +1,13 @@ import os.path from concurrent.futures import ProcessPoolExecutor -import numpy as np -import deepdanbooru as dd -import tensorflow as tf def _load_tf_and_return_tags(pil_image, threshold): + import deepdanbooru as dd + import tensorflow as tf + import numpy as np + this_folder = os.path.dirname(__file__) model_path = os.path.join(this_folder, '..', 'models', 'deepbooru', 'deepdanbooru-v3-20211112-sgd-e28') |