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

Prompt Weighting Toggle for sd_tag_batch.py 

Added a toggle to enable or disable the prompt weighting. Prompt weighting is enabled by default, which is the vanilla behavior.

I found that prompt weighting sometimes caused problems and thought that it would be nice if users could choose whether interrogator prompts were weighted or not.
pull/5/head
RookHyena 2024-05-06 20:25:34 -07:00 committed by GitHub
parent cb9ab96ad3
commit 153f3f3863
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 16 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
scripts/sd_tag_batch.py
View File

@ -26,10 +26,11 @@ class Script(scripts.Script):
prompt_weight = gr.Slider(
0.0, 1.0, value=0.5, step=0.1, label="interrogator weight"
)
use_weight = gr.Checkbox(label="Use Weighted Prompt", value=True)
use_deepbooru = gr.Checkbox(label="Use deepbooru", value=True)
return [in_front, prompt_weight, use_deepbooru]
return [in_front, prompt_weight, use_deepbooru, use_weight]
def run(self, p, in_front, prompt_weight, use_deepbooru):
def run(self, p, in_front, prompt_weight, use_deepbooru, use_weight):
# :|
if in_front:
@ -50,12 +51,20 @@ class Script(scripts.Script):
Script.interrogator = prompt
p.prompt = ""
if Script.original_prompt in ["Some", ""]:
p.prompt = Script.interrogator
elif in_front:
p.prompt = f"{Script.original_prompt}, ({Script.interrogator}:{prompt_weight})"
if use_weight:
if Script.original_prompt in ["Some", ""]:
p.prompt = Script.interrogator
elif in_front:
p.prompt = f"{Script.original_prompt}, ({Script.interrogator}:{prompt_weight})"
else:
p.prompt = f"({Script.interrogator}:{prompt_weight}), {Script.original_prompt}"
else:
p.prompt = f"({Script.interrogator}:{prompt_weight}), {Script.original_prompt}"
if Script.original_prompt in ["Some", ""]:
p.prompt = Script.interrogator
elif in_front:
p.prompt = f"{Script.original_prompt}, {Script.interrogator}"
else:
p.prompt = f"{Script.interrogator}, {Script.original_prompt}"
print(f"Prompt: {p.prompt}")
return process_images(p)