img2img batch PNG info model hash
This commit is contained in:
parent
935d9d899c
commit
4130e5db3d
|
@ -10,6 +10,7 @@ from modules import images as imgutil
|
||||||
from modules.generation_parameters_copypaste import create_override_settings_dict, parse_generation_parameters
|
from modules.generation_parameters_copypaste import create_override_settings_dict, parse_generation_parameters
|
||||||
from modules.processing import Processed, StableDiffusionProcessingImg2Img, process_images
|
from modules.processing import Processed, StableDiffusionProcessingImg2Img, process_images
|
||||||
from modules.shared import opts, state
|
from modules.shared import opts, state
|
||||||
|
from modules.sd_models import get_closet_checkpoint_match
|
||||||
import modules.shared as shared
|
import modules.shared as shared
|
||||||
import modules.processing as processing
|
import modules.processing as processing
|
||||||
from modules.ui import plaintext_to_html
|
from modules.ui import plaintext_to_html
|
||||||
|
@ -41,7 +42,8 @@ def process_batch(p, input_dir, output_dir, inpaint_mask_dir, args, to_scale=Fal
|
||||||
cfg_scale = p.cfg_scale
|
cfg_scale = p.cfg_scale
|
||||||
sampler_name = p.sampler_name
|
sampler_name = p.sampler_name
|
||||||
steps = p.steps
|
steps = p.steps
|
||||||
|
override_settings = p.override_settings
|
||||||
|
sd_model_checkpoint_override = get_closet_checkpoint_match(override_settings.get("sd_model_checkpoint", None))
|
||||||
for i, image in enumerate(images):
|
for i, image in enumerate(images):
|
||||||
state.job = f"{i+1} out of {len(images)}"
|
state.job = f"{i+1} out of {len(images)}"
|
||||||
if state.skipped:
|
if state.skipped:
|
||||||
|
@ -104,6 +106,14 @@ def process_batch(p, input_dir, output_dir, inpaint_mask_dir, args, to_scale=Fal
|
||||||
p.sampler_name = parsed_parameters.get("Sampler", sampler_name)
|
p.sampler_name = parsed_parameters.get("Sampler", sampler_name)
|
||||||
p.steps = int(parsed_parameters.get("Steps", steps))
|
p.steps = int(parsed_parameters.get("Steps", steps))
|
||||||
|
|
||||||
|
model_info = get_closet_checkpoint_match(parsed_parameters.get("Model hash", None))
|
||||||
|
if model_info is not None:
|
||||||
|
p.override_settings['sd_model_checkpoint'] = model_info.name
|
||||||
|
elif sd_model_checkpoint_override:
|
||||||
|
p.override_settings['sd_model_checkpoint'] = sd_model_checkpoint_override
|
||||||
|
else:
|
||||||
|
p.override_settings.pop("sd_model_checkpoint", None)
|
||||||
|
|
||||||
proc = modules.scripts.scripts_img2img.run(p, *args)
|
proc = modules.scripts.scripts_img2img.run(p, *args)
|
||||||
if proc is None:
|
if proc is None:
|
||||||
if output_dir:
|
if output_dir:
|
||||||
|
|
|
@ -614,7 +614,7 @@ def create_ui():
|
||||||
with gr.Accordion("PNG info", open=False):
|
with gr.Accordion("PNG info", open=False):
|
||||||
img2img_batch_use_png_info = gr.Checkbox(label="Append png info to prompts", **shared.hide_dirs, elem_id="img2img_batch_use_png_info")
|
img2img_batch_use_png_info = gr.Checkbox(label="Append png info to prompts", **shared.hide_dirs, elem_id="img2img_batch_use_png_info")
|
||||||
img2img_batch_png_info_dir = gr.Textbox(label="PNG info directory", **shared.hide_dirs, placeholder="Leave empty to use input directory", elem_id="img2img_batch_png_info_dir")
|
img2img_batch_png_info_dir = gr.Textbox(label="PNG info directory", **shared.hide_dirs, placeholder="Leave empty to use input directory", elem_id="img2img_batch_png_info_dir")
|
||||||
img2img_batch_png_info_props = gr.CheckboxGroup(["Prompt", "Negative prompt", "Seed", "CFG scale", "Sampler", "Steps"], label="Parameters to take from png info", info="Prompts from png info will be appended to prompts set in ui.")
|
img2img_batch_png_info_props = gr.CheckboxGroup(["Prompt", "Negative prompt", "Seed", "CFG scale", "Sampler", "Steps", "Model hash"], label="Parameters to take from png info", info="Prompts from png info will be appended to prompts set in ui.")
|
||||||
|
|
||||||
img2img_tabs = [tab_img2img, tab_sketch, tab_inpaint, tab_inpaint_color, tab_inpaint_upload, tab_batch]
|
img2img_tabs = [tab_img2img, tab_sketch, tab_inpaint, tab_inpaint_color, tab_inpaint_upload, tab_batch]
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue