diff options
| author | AUTOMATIC <16777216c@gmail.com> | 2022-11-20 13:39:20 +0000 |
|---|---|---|
| committer | AUTOMATIC <16777216c@gmail.com> | 2022-11-20 13:39:20 +0000 |
| commit | c81d440d876dfd2ab3560410f37442ef56fc6632 (patch) | |
| tree | 338a1c12e7d6aa0cada63de74e31be2115c907d7 /modules/ui.py | |
| parent | 47a44c7e421b98ca07e92dbf88769b04c9e28f86 (diff) | |
| download | stable-diffusion-webui-gfx803-c81d440d876dfd2ab3560410f37442ef56fc6632.tar.gz stable-diffusion-webui-gfx803-c81d440d876dfd2ab3560410f37442ef56fc6632.tar.bz2 stable-diffusion-webui-gfx803-c81d440d876dfd2ab3560410f37442ef56fc6632.zip | |
moved deepdanbooru to pure pytorch implementation
Diffstat (limited to 'modules/ui.py')
| -rw-r--r-- | modules/ui.py | 7 |
1 files changed, 2 insertions, 5 deletions
diff --git a/modules/ui.py b/modules/ui.py index a5953fce..e6da1b2a 100644 --- a/modules/ui.py +++ b/modules/ui.py @@ -19,14 +19,11 @@ import numpy as np from PIL import Image, PngImagePlugin
-from modules import sd_hijack, sd_models, localization, script_callbacks, ui_extensions
+from modules import sd_hijack, sd_models, localization, script_callbacks, ui_extensions, deepbooru
from modules.paths import script_path
from modules.shared import opts, cmd_opts, restricted_opts
-if cmd_opts.deepdanbooru:
- from modules.deepbooru import get_deepbooru_tags
-
import modules.codeformer_model
import modules.generation_parameters_copypaste as parameters_copypaste
import modules.gfpgan_model
@@ -352,7 +349,7 @@ def interrogate(image): def interrogate_deepbooru(image):
- prompt = get_deepbooru_tags(image)
+ prompt = deepbooru.model.tag(image)
return gr_show(True) if prompt is None else prompt
|