1580158
/
sd-Img2img-batch-interrogator
mirror of https://github.com/Alvi-alvarez/sd-Img2img-batch-interrogator.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix prompt

pull/5/head
Alvi 2023-08-30 20:49:42 -03:00
parent 3db3b37551
commit 7f4e4eb064
1 changed files with 4 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
scripts/sd_tag_batch.py
View File

@ -16,11 +16,11 @@ class Script(scripts.Script):
return is_img2img
def ui(self, is_img2img):
in_front = gr.Checkbox(label="Prompt in front")
in_front = gr.Checkbox(label="Prompt in front", value=True)
prompt_weight = gr.Slider(
0.0, 1.0, value=0.5, step=0.1, label="interrogator weight"
)
use_deepbooru = gr.Checkbox(label="Use deepbooru")
use_deepbooru = gr.Checkbox(label="Use deepbooru", value=True)
return [in_front, prompt_weight, use_deepbooru]
def run(self, p, in_front, prompt_weight, use_deepbooru):
@ -30,9 +30,9 @@ class Script(scripts.Script):
_check = f"{Script.orginal_prompt}, ({Script.interrogator}:{prompt_weight})"
else:
_check = f"({Script.interrogator}:{prompt_weight}), {Script.orginal_prompt}"
if p.prompt != _check:
if p.prompt not in [_check, Script.interrogator]:
Script.orginal_prompt = p.prompt
if use_deepbooru:
prompt = deepbooru.model.tag(p.init_images[0])
@ -40,7 +40,6 @@ class Script(scripts.Script):
prompt = shared.interrogator.interrogate(p.init_images[0])
Script.interrogator = prompt
p.prompt = ""
if Script.orginal_prompt in ["Some", ""]:
p.prompt = Script.interrogator