hoelee/stable-diffusion-webui
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2025-08-08 13:19:54 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

moved deepdanbooru to pure pytorch implementation

Browse Source
This commit is contained in:
AUTOMATIC
2022-11-20 16:39:20 +03:00
parent 47a44c7e42
commit c81d440d87
8 changed files with 757 additions and 175 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
modules/ui.py
View File

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