Refactor into single package
Aarni Koskela
parent
f60522578c
commit
0320a42f6e
|
|
@ -1,442 +1,5 @@
|
|||
from __future__ import annotations
|
||||
# Automatic1111 entry point.
|
||||
|
||||
import logging
|
||||
import math
|
||||
from pathlib import Path
|
||||
from string import Template
|
||||
from sd_dynamic_prompts.dynamic_prompting import Script
|
||||
|
||||
import gradio as gr
|
||||
import modules.scripts as scripts
|
||||
from dynamicprompts.generators.promptgenerator import GeneratorException
|
||||
from dynamicprompts.wildcardmanager import WildcardManager
|
||||
from modules.devices import get_optimal_device
|
||||
from modules.processing import fix_seed
|
||||
from modules.shared import opts
|
||||
|
||||
from prompts import prompt_writer
|
||||
from prompts.consts import MAGIC_PROMPT_MODELS
|
||||
from prompts.generator_builder import GeneratorBuilder
|
||||
from prompts.uicreation import UiCreation
|
||||
from prompts.utils import get_unique_path, slugify
|
||||
from ui import save_params, settings, wildcards_tab
|
||||
|
||||
VERSION = "2.5.2"
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
logger.setLevel(logging.INFO)
|
||||
|
||||
is_debug = getattr(opts, "is_debug", False)
|
||||
|
||||
if is_debug:
|
||||
logger.setLevel(logging.DEBUG)
|
||||
|
||||
base_dir = Path(scripts.basedir())
|
||||
|
||||
|
||||
def get_wildcard_manager():
|
||||
wildcard_dir = getattr(opts, "wildcard_dir", None)
|
||||
if wildcard_dir is None:
|
||||
wildcard_dir = base_dir / "wildcards"
|
||||
else:
|
||||
wildcard_dir = Path(wildcard_dir)
|
||||
|
||||
wildcard_manager = WildcardManager(wildcard_dir)
|
||||
wildcard_manager.ensure_directory()
|
||||
|
||||
return wildcard_manager
|
||||
|
||||
|
||||
def get_prompts(p):
|
||||
original_prompt = p.all_prompts[0] if len(p.all_prompts) > 0 else p.prompt
|
||||
original_negative_prompt = (
|
||||
p.all_negative_prompts[0]
|
||||
if len(p.all_negative_prompts) > 0
|
||||
else p.negative_prompt
|
||||
)
|
||||
|
||||
return original_prompt, original_negative_prompt
|
||||
|
||||
|
||||
def get_seeds(p, num_seeds, use_fixed_seed):
|
||||
if use_fixed_seed:
|
||||
all_seeds = [p.seed] * num_seeds
|
||||
else:
|
||||
all_seeds = [
|
||||
int(p.seed) + (x if p.subseed_strength == 0 else 0)
|
||||
for x in range(num_seeds)
|
||||
]
|
||||
|
||||
return all_seeds
|
||||
|
||||
|
||||
wildcard_manager = get_wildcard_manager()
|
||||
wildcards_tab.initialize(wildcard_manager)
|
||||
save_params.initialize()
|
||||
settings.initialize()
|
||||
|
||||
device = 0 if get_optimal_device() == "cuda" else -1
|
||||
|
||||
|
||||
def generate_prompts(
|
||||
prompt_generator,
|
||||
negative_prompt_generator,
|
||||
prompt,
|
||||
negative_prompt,
|
||||
num_prompts,
|
||||
):
|
||||
all_prompts = prompt_generator.generate(prompt, num_prompts)
|
||||
total_prompts = len(all_prompts)
|
||||
|
||||
all_negative_prompts = negative_prompt_generator.generate(
|
||||
negative_prompt,
|
||||
num_prompts,
|
||||
)
|
||||
|
||||
if len(all_negative_prompts) < total_prompts:
|
||||
all_negative_prompts = all_negative_prompts * (
|
||||
total_prompts // len(all_negative_prompts) + 1
|
||||
)
|
||||
|
||||
all_negative_prompts = all_negative_prompts[:total_prompts]
|
||||
|
||||
return all_prompts, all_negative_prompts
|
||||
|
||||
|
||||
class Script(scripts.Script):
|
||||
def title(self):
|
||||
return f"Dynamic Prompts v{VERSION}"
|
||||
|
||||
def show(self, is_img2img):
|
||||
return scripts.AlwaysVisible
|
||||
|
||||
def ui(self, is_img2img):
|
||||
ui_creation = UiCreation(wildcard_manager)
|
||||
wildcard_html = ui_creation.probe()
|
||||
|
||||
html_path = base_dir / "helptext.html"
|
||||
html = html_path.open().read()
|
||||
html = Template(html).substitute(
|
||||
wildcard_html=wildcard_html,
|
||||
WILDCARD_DIR=wildcard_manager.path,
|
||||
VERSION=VERSION,
|
||||
)
|
||||
|
||||
jinja_html_path = base_dir / "jinja_help.html"
|
||||
jinja_help = jinja_html_path.open().read()
|
||||
|
||||
with gr.Group(elem_id="dynamic-prompting"):
|
||||
with gr.Accordion("Dynamic Prompts", open=False):
|
||||
is_enabled = gr.Checkbox(
|
||||
label="Dynamic Prompts enabled",
|
||||
value=True,
|
||||
elem_id="dynamic-prompts-enabled",
|
||||
)
|
||||
|
||||
with gr.Group():
|
||||
is_combinatorial = gr.Checkbox(
|
||||
label="Combinatorial generation",
|
||||
value=False,
|
||||
elem_id="is-combinatorial",
|
||||
)
|
||||
|
||||
max_generations = gr.Slider(
|
||||
label="Max generations (0 = all combinations - the batch count value is ignored)",
|
||||
minimum=0,
|
||||
maximum=1000,
|
||||
step=1,
|
||||
value=0,
|
||||
elem_id="max-generations",
|
||||
)
|
||||
|
||||
combinatorial_batches = gr.Slider(
|
||||
label="Combinatorial batches",
|
||||
minimum=1,
|
||||
maximum=10,
|
||||
step=1,
|
||||
value=1,
|
||||
elem_id="combinatorial-times",
|
||||
)
|
||||
|
||||
with gr.Accordion("Prompt Magic", open=False):
|
||||
with gr.Group():
|
||||
is_magic_prompt = gr.Checkbox(
|
||||
label="Magic prompt",
|
||||
value=False,
|
||||
elem_id="is-magicprompt",
|
||||
)
|
||||
|
||||
magic_prompt_length = gr.Slider(
|
||||
label="Max magic prompt length",
|
||||
value=100,
|
||||
minimum=30,
|
||||
maximum=300,
|
||||
step=10,
|
||||
)
|
||||
|
||||
magic_temp_value = gr.Slider(
|
||||
label="Magic prompt creativity",
|
||||
value=0.7,
|
||||
minimum=0.1,
|
||||
maximum=3.0,
|
||||
step=0.10,
|
||||
)
|
||||
|
||||
magic_model = gr.Dropdown(
|
||||
MAGIC_PROMPT_MODELS,
|
||||
value=MAGIC_PROMPT_MODELS[0],
|
||||
multiselect=False,
|
||||
label="Magic prompt model",
|
||||
elem_id="magic-prompt-model",
|
||||
)
|
||||
|
||||
magic_blocklist_regex = gr.Textbox(
|
||||
label="Magic prompt blocklist regex",
|
||||
value="",
|
||||
elem_id="magic-prompt-blocklist-regex",
|
||||
placeholder=(
|
||||
"Regular expression pattern for blocking terms out of the generated prompt. Applied case-insensitively. "
|
||||
'For instance, to block both "purple" and "interdimensional", you could use the pattern "purple|interdimensional".'
|
||||
),
|
||||
)
|
||||
|
||||
is_feeling_lucky = gr.Checkbox(
|
||||
label="I'm feeling lucky",
|
||||
value=False,
|
||||
elem_id="is-feelinglucky",
|
||||
)
|
||||
|
||||
with gr.Group():
|
||||
is_attention_grabber = gr.Checkbox(
|
||||
label="Attention grabber",
|
||||
value=False,
|
||||
elem_id="is-attention-grabber",
|
||||
)
|
||||
|
||||
min_attention = gr.Slider(
|
||||
label="Minimum attention",
|
||||
value=1.1,
|
||||
minimum=-1,
|
||||
maximum=2,
|
||||
step=0.1,
|
||||
)
|
||||
|
||||
max_attention = gr.Slider(
|
||||
label="Maximum attention",
|
||||
value=1.5,
|
||||
minimum=-1,
|
||||
maximum=2,
|
||||
step=0.1,
|
||||
)
|
||||
|
||||
disable_negative_prompt = gr.Checkbox(
|
||||
label="Don't apply to negative prompts",
|
||||
value=True,
|
||||
elem_id="disable-negative-prompt",
|
||||
)
|
||||
|
||||
with gr.Accordion("Need help?", open=False):
|
||||
gr.HTML(html)
|
||||
|
||||
with gr.Group():
|
||||
with gr.Accordion("Jinja2 templates", open=False):
|
||||
enable_jinja_templates = gr.Checkbox(
|
||||
label="Enable Jinja2 templates",
|
||||
value=False,
|
||||
elem_id="enable-jinja-templates",
|
||||
)
|
||||
|
||||
with gr.Accordion("Help for Jinja2 templates", open=False):
|
||||
gr.HTML(jinja_help)
|
||||
|
||||
with gr.Group():
|
||||
with gr.Accordion("Advanced options", open=False):
|
||||
gr.HTML(
|
||||
"Some settings have been moved to the settings tab. Find them in the Dynamic Prompts section.",
|
||||
)
|
||||
|
||||
unlink_seed_from_prompt = gr.Checkbox(
|
||||
label="Unlink seed from prompt",
|
||||
value=False,
|
||||
elem_id="unlink-seed-from-prompt",
|
||||
)
|
||||
|
||||
use_fixed_seed = gr.Checkbox(
|
||||
label="Fixed seed",
|
||||
value=False,
|
||||
elem_id="is-fixed-seed",
|
||||
)
|
||||
|
||||
gr.Checkbox(
|
||||
label="Write raw prompt to image",
|
||||
value=False,
|
||||
visible=False, # For some reason, removing this line causes Auto1111 to hang
|
||||
elem_id="write-raw-template",
|
||||
)
|
||||
|
||||
no_image_generation = gr.Checkbox(
|
||||
label="Don't generate images",
|
||||
value=False,
|
||||
elem_id="no-image-generation",
|
||||
)
|
||||
|
||||
gr.Checkbox(
|
||||
label="Write prompts to file",
|
||||
value=False,
|
||||
elem_id="write-prompts",
|
||||
visible=False, # For some reason, removing this line causes Auto1111 to hang
|
||||
)
|
||||
|
||||
return [
|
||||
is_enabled,
|
||||
is_combinatorial,
|
||||
combinatorial_batches,
|
||||
is_magic_prompt,
|
||||
is_feeling_lucky,
|
||||
is_attention_grabber,
|
||||
min_attention,
|
||||
max_attention,
|
||||
magic_prompt_length,
|
||||
magic_temp_value,
|
||||
use_fixed_seed,
|
||||
unlink_seed_from_prompt,
|
||||
disable_negative_prompt,
|
||||
enable_jinja_templates,
|
||||
no_image_generation,
|
||||
max_generations,
|
||||
magic_model,
|
||||
magic_blocklist_regex,
|
||||
]
|
||||
|
||||
def process(
|
||||
self,
|
||||
p,
|
||||
is_enabled,
|
||||
is_combinatorial,
|
||||
combinatorial_batches,
|
||||
is_magic_prompt,
|
||||
is_feeling_lucky,
|
||||
is_attention_grabber,
|
||||
min_attention,
|
||||
max_attention,
|
||||
magic_prompt_length,
|
||||
magic_temp_value,
|
||||
use_fixed_seed,
|
||||
unlink_seed_from_prompt,
|
||||
disable_negative_prompt,
|
||||
enable_jinja_templates,
|
||||
no_image_generation,
|
||||
max_generations,
|
||||
magic_model,
|
||||
magic_blocklist_regex: str | None,
|
||||
):
|
||||
|
||||
if not is_enabled:
|
||||
logger.debug("Dynamic prompts disabled - exiting")
|
||||
return p
|
||||
|
||||
self._p = p
|
||||
|
||||
ignore_whitespace = opts.dp_ignore_whitespace
|
||||
write_prompts = opts.dp_write_prompts_to_file
|
||||
|
||||
fix_seed(p)
|
||||
|
||||
original_prompt, original_negative_prompt = get_prompts(p)
|
||||
original_seed = p.seed
|
||||
num_images = p.n_iter * p.batch_size
|
||||
|
||||
if is_combinatorial:
|
||||
if max_generations == 0:
|
||||
num_images = None
|
||||
else:
|
||||
num_images = max_generations
|
||||
|
||||
combinatorial_batches = int(combinatorial_batches)
|
||||
|
||||
try:
|
||||
logger.debug("Creating generator")
|
||||
generator_builder = (
|
||||
GeneratorBuilder(wildcard_manager, ignore_whitespace=ignore_whitespace)
|
||||
.set_is_feeling_lucky(is_feeling_lucky)
|
||||
.set_is_attention_grabber(
|
||||
is_attention_grabber,
|
||||
min_attention,
|
||||
max_attention,
|
||||
)
|
||||
.set_is_jinja_template(enable_jinja_templates)
|
||||
.set_is_combinatorial(is_combinatorial, combinatorial_batches)
|
||||
.set_is_magic_prompt(
|
||||
is_magic_prompt,
|
||||
magic_model=magic_model,
|
||||
magic_prompt_length=magic_prompt_length,
|
||||
magic_temp_value=magic_temp_value,
|
||||
magic_blocklist_regex=magic_blocklist_regex,
|
||||
)
|
||||
.set_is_dummy(False)
|
||||
.set_unlink_seed_from_prompt(unlink_seed_from_prompt)
|
||||
.set_seed(original_seed)
|
||||
.set_context(p)
|
||||
)
|
||||
|
||||
generator = generator_builder.create_generator()
|
||||
|
||||
if disable_negative_prompt:
|
||||
generator_builder.disable_prompt_magic()
|
||||
negative_generator = generator_builder.create_generator()
|
||||
else:
|
||||
negative_generator = generator
|
||||
|
||||
all_prompts, all_negative_prompts = generate_prompts(
|
||||
generator,
|
||||
negative_generator,
|
||||
original_prompt,
|
||||
original_negative_prompt,
|
||||
num_images,
|
||||
)
|
||||
|
||||
except GeneratorException as e:
|
||||
logger.exception(e)
|
||||
all_prompts = [str(e)]
|
||||
all_negative_prompts = [str(e)]
|
||||
|
||||
updated_count = len(all_prompts)
|
||||
p.n_iter = math.ceil(updated_count / p.batch_size)
|
||||
|
||||
all_seeds = get_seeds(p, updated_count, use_fixed_seed)
|
||||
|
||||
logger.info(
|
||||
f"Prompt matrix will create {updated_count} images in a total of {p.n_iter} batches.",
|
||||
)
|
||||
|
||||
try:
|
||||
|
||||
if write_prompts:
|
||||
prompt_filename = get_unique_path(
|
||||
Path(p.outpath_samples),
|
||||
slugify(original_prompt),
|
||||
suffix="csv",
|
||||
)
|
||||
prompt_writer.write_prompts(
|
||||
prompt_filename,
|
||||
original_prompt,
|
||||
original_negative_prompt,
|
||||
all_prompts,
|
||||
all_negative_prompts,
|
||||
)
|
||||
except Exception as e:
|
||||
logger.error(f"Failed to write prompts to file: {e}")
|
||||
|
||||
p.all_prompts = all_prompts
|
||||
p.all_negative_prompts = all_negative_prompts
|
||||
if no_image_generation:
|
||||
logger.debug("No image generation requested - exiting")
|
||||
# Need a minimum of batch size images to avoid errors
|
||||
p.batch_size = 1
|
||||
p.all_prompts = all_prompts[0:1]
|
||||
|
||||
p.all_seeds = all_seeds
|
||||
|
||||
p.prompt_for_display = original_prompt
|
||||
|
||||
p.prompt = original_prompt
|
||||
p.seed = original_seed
|
||||
__all__ = ["Script"]
|
||||
|
|
|
|||
|
|
@ -17,3 +17,4 @@ MAGIC_PROMPT_MODELS = [
|
|||
"DrishtiSharma/StableDiffusion-Prompt-Generator-GPT-Neo-125M",
|
||||
]
|
||||
DEFAULT_MAGIC_MODEL = MAGIC_PROMPT_MODELS[0]
|
||||
OPTION_WRITE_RAW_TEMPLATE = "pp_write_raw_template"
|
||||
|
|
@ -0,0 +1,442 @@
|
|||
from __future__ import annotations
|
||||
|
||||
import logging
|
||||
import math
|
||||
from pathlib import Path
|
||||
from string import Template
|
||||
|
||||
import gradio as gr
|
||||
import modules.scripts as scripts
|
||||
from dynamicprompts.generators.promptgenerator import GeneratorException
|
||||
from dynamicprompts.wildcardmanager import WildcardManager
|
||||
from modules.devices import get_optimal_device
|
||||
from modules.processing import fix_seed
|
||||
from modules.shared import opts
|
||||
|
||||
from sd_dynamic_prompts import prompt_writer
|
||||
from sd_dynamic_prompts.consts import MAGIC_PROMPT_MODELS
|
||||
from sd_dynamic_prompts.generator_builder import GeneratorBuilder
|
||||
from sd_dynamic_prompts.ui import save_params, settings, wildcards_tab
|
||||
from sd_dynamic_prompts.ui.uicreation import UiCreation
|
||||
from sd_dynamic_prompts.utils import get_unique_path, slugify
|
||||
|
||||
VERSION = "2.5.2"
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
logger.setLevel(logging.INFO)
|
||||
|
||||
is_debug = getattr(opts, "is_debug", False)
|
||||
|
||||
if is_debug:
|
||||
logger.setLevel(logging.DEBUG)
|
||||
|
||||
base_dir = Path(scripts.basedir())
|
||||
|
||||
|
||||
def get_wildcard_manager():
|
||||
wildcard_dir = getattr(opts, "wildcard_dir", None)
|
||||
if wildcard_dir is None:
|
||||
wildcard_dir = base_dir / "wildcards"
|
||||
else:
|
||||
wildcard_dir = Path(wildcard_dir)
|
||||
|
||||
wildcard_manager = WildcardManager(wildcard_dir)
|
||||
wildcard_manager.ensure_directory()
|
||||
|
||||
return wildcard_manager
|
||||
|
||||
|
||||
def get_prompts(p):
|
||||
original_prompt = p.all_prompts[0] if len(p.all_prompts) > 0 else p.prompt
|
||||
original_negative_prompt = (
|
||||
p.all_negative_prompts[0]
|
||||
if len(p.all_negative_prompts) > 0
|
||||
else p.negative_prompt
|
||||
)
|
||||
|
||||
return original_prompt, original_negative_prompt
|
||||
|
||||
|
||||
def get_seeds(p, num_seeds, use_fixed_seed):
|
||||
if use_fixed_seed:
|
||||
all_seeds = [p.seed] * num_seeds
|
||||
else:
|
||||
all_seeds = [
|
||||
int(p.seed) + (x if p.subseed_strength == 0 else 0)
|
||||
for x in range(num_seeds)
|
||||
]
|
||||
|
||||
return all_seeds
|
||||
|
||||
|
||||
wildcard_manager = get_wildcard_manager()
|
||||
wildcards_tab.initialize(wildcard_manager)
|
||||
save_params.initialize()
|
||||
settings.initialize()
|
||||
|
||||
device = 0 if get_optimal_device() == "cuda" else -1
|
||||
|
||||
|
||||
def generate_prompts(
|
||||
prompt_generator,
|
||||
negative_prompt_generator,
|
||||
prompt,
|
||||
negative_prompt,
|
||||
num_prompts,
|
||||
):
|
||||
all_prompts = prompt_generator.generate(prompt, num_prompts)
|
||||
total_prompts = len(all_prompts)
|
||||
|
||||
all_negative_prompts = negative_prompt_generator.generate(
|
||||
negative_prompt,
|
||||
num_prompts,
|
||||
)
|
||||
|
||||
if len(all_negative_prompts) < total_prompts:
|
||||
all_negative_prompts = all_negative_prompts * (
|
||||
total_prompts // len(all_negative_prompts) + 1
|
||||
)
|
||||
|
||||
all_negative_prompts = all_negative_prompts[:total_prompts]
|
||||
|
||||
return all_prompts, all_negative_prompts
|
||||
|
||||
|
||||
class Script(scripts.Script):
|
||||
def title(self):
|
||||
return f"Dynamic Prompts v{VERSION}"
|
||||
|
||||
def show(self, is_img2img):
|
||||
return scripts.AlwaysVisible
|
||||
|
||||
def ui(self, is_img2img):
|
||||
ui_creation = UiCreation(wildcard_manager)
|
||||
wildcard_html = ui_creation.probe()
|
||||
|
||||
html_path = base_dir / "helptext.html"
|
||||
html = html_path.open().read()
|
||||
html = Template(html).substitute(
|
||||
wildcard_html=wildcard_html,
|
||||
WILDCARD_DIR=wildcard_manager.path,
|
||||
VERSION=VERSION,
|
||||
)
|
||||
|
||||
jinja_html_path = base_dir / "jinja_help.html"
|
||||
jinja_help = jinja_html_path.open().read()
|
||||
|
||||
with gr.Group(elem_id="dynamic-prompting"):
|
||||
with gr.Accordion("Dynamic Prompts", open=False):
|
||||
is_enabled = gr.Checkbox(
|
||||
label="Dynamic Prompts enabled",
|
||||
value=True,
|
||||
elem_id="dynamic-prompts-enabled",
|
||||
)
|
||||
|
||||
with gr.Group():
|
||||
is_combinatorial = gr.Checkbox(
|
||||
label="Combinatorial generation",
|
||||
value=False,
|
||||
elem_id="is-combinatorial",
|
||||
)
|
||||
|
||||
max_generations = gr.Slider(
|
||||
label="Max generations (0 = all combinations - the batch count value is ignored)",
|
||||
minimum=0,
|
||||
maximum=1000,
|
||||
step=1,
|
||||
value=0,
|
||||
elem_id="max-generations",
|
||||
)
|
||||
|
||||
combinatorial_batches = gr.Slider(
|
||||
label="Combinatorial batches",
|
||||
minimum=1,
|
||||
maximum=10,
|
||||
step=1,
|
||||
value=1,
|
||||
elem_id="combinatorial-times",
|
||||
)
|
||||
|
||||
with gr.Accordion("Prompt Magic", open=False):
|
||||
with gr.Group():
|
||||
is_magic_prompt = gr.Checkbox(
|
||||
label="Magic prompt",
|
||||
value=False,
|
||||
elem_id="is-magicprompt",
|
||||
)
|
||||
|
||||
magic_prompt_length = gr.Slider(
|
||||
label="Max magic prompt length",
|
||||
value=100,
|
||||
minimum=30,
|
||||
maximum=300,
|
||||
step=10,
|
||||
)
|
||||
|
||||
magic_temp_value = gr.Slider(
|
||||
label="Magic prompt creativity",
|
||||
value=0.7,
|
||||
minimum=0.1,
|
||||
maximum=3.0,
|
||||
step=0.10,
|
||||
)
|
||||
|
||||
magic_model = gr.Dropdown(
|
||||
MAGIC_PROMPT_MODELS,
|
||||
value=MAGIC_PROMPT_MODELS[0],
|
||||
multiselect=False,
|
||||
label="Magic prompt model",
|
||||
elem_id="magic-prompt-model",
|
||||
)
|
||||
|
||||
magic_blocklist_regex = gr.Textbox(
|
||||
label="Magic prompt blocklist regex",
|
||||
value="",
|
||||
elem_id="magic-prompt-blocklist-regex",
|
||||
placeholder=(
|
||||
"Regular expression pattern for blocking terms out of the generated prompt. Applied case-insensitively. "
|
||||
'For instance, to block both "purple" and "interdimensional", you could use the pattern "purple|interdimensional".'
|
||||
),
|
||||
)
|
||||
|
||||
is_feeling_lucky = gr.Checkbox(
|
||||
label="I'm feeling lucky",
|
||||
value=False,
|
||||
elem_id="is-feelinglucky",
|
||||
)
|
||||
|
||||
with gr.Group():
|
||||
is_attention_grabber = gr.Checkbox(
|
||||
label="Attention grabber",
|
||||
value=False,
|
||||
elem_id="is-attention-grabber",
|
||||
)
|
||||
|
||||
min_attention = gr.Slider(
|
||||
label="Minimum attention",
|
||||
value=1.1,
|
||||
minimum=-1,
|
||||
maximum=2,
|
||||
step=0.1,
|
||||
)
|
||||
|
||||
max_attention = gr.Slider(
|
||||
label="Maximum attention",
|
||||
value=1.5,
|
||||
minimum=-1,
|
||||
maximum=2,
|
||||
step=0.1,
|
||||
)
|
||||
|
||||
disable_negative_prompt = gr.Checkbox(
|
||||
label="Don't apply to negative prompts",
|
||||
value=True,
|
||||
elem_id="disable-negative-prompt",
|
||||
)
|
||||
|
||||
with gr.Accordion("Need help?", open=False):
|
||||
gr.HTML(html)
|
||||
|
||||
with gr.Group():
|
||||
with gr.Accordion("Jinja2 templates", open=False):
|
||||
enable_jinja_templates = gr.Checkbox(
|
||||
label="Enable Jinja2 templates",
|
||||
value=False,
|
||||
elem_id="enable-jinja-templates",
|
||||
)
|
||||
|
||||
with gr.Accordion("Help for Jinja2 templates", open=False):
|
||||
gr.HTML(jinja_help)
|
||||
|
||||
with gr.Group():
|
||||
with gr.Accordion("Advanced options", open=False):
|
||||
gr.HTML(
|
||||
"Some settings have been moved to the settings tab. Find them in the Dynamic Prompts section.",
|
||||
)
|
||||
|
||||
unlink_seed_from_prompt = gr.Checkbox(
|
||||
label="Unlink seed from prompt",
|
||||
value=False,
|
||||
elem_id="unlink-seed-from-prompt",
|
||||
)
|
||||
|
||||
use_fixed_seed = gr.Checkbox(
|
||||
label="Fixed seed",
|
||||
value=False,
|
||||
elem_id="is-fixed-seed",
|
||||
)
|
||||
|
||||
gr.Checkbox(
|
||||
label="Write raw prompt to image",
|
||||
value=False,
|
||||
visible=False, # For some reason, removing this line causes Auto1111 to hang
|
||||
elem_id="write-raw-template",
|
||||
)
|
||||
|
||||
no_image_generation = gr.Checkbox(
|
||||
label="Don't generate images",
|
||||
value=False,
|
||||
elem_id="no-image-generation",
|
||||
)
|
||||
|
||||
gr.Checkbox(
|
||||
label="Write prompts to file",
|
||||
value=False,
|
||||
elem_id="write-prompts",
|
||||
visible=False, # For some reason, removing this line causes Auto1111 to hang
|
||||
)
|
||||
|
||||
return [
|
||||
is_enabled,
|
||||
is_combinatorial,
|
||||
combinatorial_batches,
|
||||
is_magic_prompt,
|
||||
is_feeling_lucky,
|
||||
is_attention_grabber,
|
||||
min_attention,
|
||||
max_attention,
|
||||
magic_prompt_length,
|
||||
magic_temp_value,
|
||||
use_fixed_seed,
|
||||
unlink_seed_from_prompt,
|
||||
disable_negative_prompt,
|
||||
enable_jinja_templates,
|
||||
no_image_generation,
|
||||
max_generations,
|
||||
magic_model,
|
||||
magic_blocklist_regex,
|
||||
]
|
||||
|
||||
def process(
|
||||
self,
|
||||
p,
|
||||
is_enabled,
|
||||
is_combinatorial,
|
||||
combinatorial_batches,
|
||||
is_magic_prompt,
|
||||
is_feeling_lucky,
|
||||
is_attention_grabber,
|
||||
min_attention,
|
||||
max_attention,
|
||||
magic_prompt_length,
|
||||
magic_temp_value,
|
||||
use_fixed_seed,
|
||||
unlink_seed_from_prompt,
|
||||
disable_negative_prompt,
|
||||
enable_jinja_templates,
|
||||
no_image_generation,
|
||||
max_generations,
|
||||
magic_model,
|
||||
magic_blocklist_regex: str | None,
|
||||
):
|
||||
|
||||
if not is_enabled:
|
||||
logger.debug("Dynamic prompts disabled - exiting")
|
||||
return p
|
||||
|
||||
self._p = p
|
||||
|
||||
ignore_whitespace = opts.dp_ignore_whitespace
|
||||
write_prompts = opts.dp_write_prompts_to_file
|
||||
|
||||
fix_seed(p)
|
||||
|
||||
original_prompt, original_negative_prompt = get_prompts(p)
|
||||
original_seed = p.seed
|
||||
num_images = p.n_iter * p.batch_size
|
||||
|
||||
if is_combinatorial:
|
||||
if max_generations == 0:
|
||||
num_images = None
|
||||
else:
|
||||
num_images = max_generations
|
||||
|
||||
combinatorial_batches = int(combinatorial_batches)
|
||||
|
||||
try:
|
||||
logger.debug("Creating generator")
|
||||
generator_builder = (
|
||||
GeneratorBuilder(wildcard_manager, ignore_whitespace=ignore_whitespace)
|
||||
.set_is_feeling_lucky(is_feeling_lucky)
|
||||
.set_is_attention_grabber(
|
||||
is_attention_grabber,
|
||||
min_attention,
|
||||
max_attention,
|
||||
)
|
||||
.set_is_jinja_template(enable_jinja_templates)
|
||||
.set_is_combinatorial(is_combinatorial, combinatorial_batches)
|
||||
.set_is_magic_prompt(
|
||||
is_magic_prompt,
|
||||
magic_model=magic_model,
|
||||
magic_prompt_length=magic_prompt_length,
|
||||
magic_temp_value=magic_temp_value,
|
||||
magic_blocklist_regex=magic_blocklist_regex,
|
||||
)
|
||||
.set_is_dummy(False)
|
||||
.set_unlink_seed_from_prompt(unlink_seed_from_prompt)
|
||||
.set_seed(original_seed)
|
||||
.set_context(p)
|
||||
)
|
||||
|
||||
generator = generator_builder.create_generator()
|
||||
|
||||
if disable_negative_prompt:
|
||||
generator_builder.disable_prompt_magic()
|
||||
negative_generator = generator_builder.create_generator()
|
||||
else:
|
||||
negative_generator = generator
|
||||
|
||||
all_prompts, all_negative_prompts = generate_prompts(
|
||||
generator,
|
||||
negative_generator,
|
||||
original_prompt,
|
||||
original_negative_prompt,
|
||||
num_images,
|
||||
)
|
||||
|
||||
except GeneratorException as e:
|
||||
logger.exception(e)
|
||||
all_prompts = [str(e)]
|
||||
all_negative_prompts = [str(e)]
|
||||
|
||||
updated_count = len(all_prompts)
|
||||
p.n_iter = math.ceil(updated_count / p.batch_size)
|
||||
|
||||
all_seeds = get_seeds(p, updated_count, use_fixed_seed)
|
||||
|
||||
logger.info(
|
||||
f"Prompt matrix will create {updated_count} images in a total of {p.n_iter} batches.",
|
||||
)
|
||||
|
||||
try:
|
||||
|
||||
if write_prompts:
|
||||
prompt_filename = get_unique_path(
|
||||
Path(p.outpath_samples),
|
||||
slugify(original_prompt),
|
||||
suffix="csv",
|
||||
)
|
||||
prompt_writer.write_prompts(
|
||||
prompt_filename,
|
||||
original_prompt,
|
||||
original_negative_prompt,
|
||||
all_prompts,
|
||||
all_negative_prompts,
|
||||
)
|
||||
except Exception as e:
|
||||
logger.error(f"Failed to write prompts to file: {e}")
|
||||
|
||||
p.all_prompts = all_prompts
|
||||
p.all_negative_prompts = all_negative_prompts
|
||||
if no_image_generation:
|
||||
logger.debug("No image generation requested - exiting")
|
||||
# Need a minimum of batch size images to avoid errors
|
||||
p.batch_size = 1
|
||||
p.all_prompts = all_prompts[0:1]
|
||||
|
||||
p.all_seeds = all_seeds
|
||||
|
||||
p.prompt_for_display = original_prompt
|
||||
|
||||
p.prompt = original_prompt
|
||||
p.seed = original_seed
|
||||
|
|
@ -14,7 +14,7 @@ from dynamicprompts.generators import (
|
|||
from dynamicprompts.generators.attentiongenerator import AttentionGenerator
|
||||
from dynamicprompts.generators.magicprompt import MagicPromptGenerator
|
||||
|
||||
from prompts.consts import DEFAULT_MAGIC_MODEL
|
||||
from sd_dynamic_prompts.consts import DEFAULT_MAGIC_MODEL
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
|
|
@ -3,7 +3,7 @@ from unittest.mock import patch
|
|||
from dynamicprompts.generators.magicprompt import MagicPromptGenerator
|
||||
from dynamicprompts.wildcardmanager import WildcardManager
|
||||
|
||||
from prompts.generator_builder import GeneratorBuilder
|
||||
from sd_dynamic_prompts.generator_builder import GeneratorBuilder
|
||||
|
||||
|
||||
def test_magic_blocklist_regexp(tmp_path):
|
||||
|
|
|
|||
|
|
@ -1,4 +1,4 @@
|
|||
from prompts.utils import slugify
|
||||
from sd_dynamic_prompts.utils import slugify
|
||||
|
||||
|
||||
class TestSlugify:
|
||||
|
|
|
|||
|
|
@ -1 +0,0 @@
|
|||
OPTION_WRITE_RAW_TEMPLATE = "pp_write_raw_template"
|
||||
Loading…
Reference in New Issue