Add support for saving init images in img2img

1 files changed, 3 insertions, 0 deletions

diff --git a/modules/shared.py b/modules/shared.py
index 5fd0eecb..69c2b21e 100644
--- a/modules/shared.py
+++ b/modules/shared.py
@@ -39,6 +39,7 @@ restricted_opts = {
     "outdir_grids",

     "outdir_txt2img_grids",

     "outdir_save",

+    "outdir_init_images"

 }

 

 ui_reorder_categories = [

@@ -253,6 +254,7 @@ options_templates.update(options_section(('saving-images', "Saving images/grids"
     "use_upscaler_name_as_suffix": OptionInfo(False, "Use upscaler name as filename suffix in the extras tab"),

     "save_selected_only": OptionInfo(True, "When using 'Save' button, only save a single selected image"),

     "do_not_add_watermark": OptionInfo(False, "Do not add watermark to images"),

+    "save_init_img": OptionInfo(True, "Save init images when using img2img"),

 

     "temp_dir":  OptionInfo("", "Directory for temporary images; leave empty for default"),

     "clean_temp_dir_at_start": OptionInfo(False, "Cleanup non-default temporary directory when starting webui"),

@@ -268,6 +270,7 @@ options_templates.update(options_section(('saving-paths', "Paths for saving"), {
     "outdir_txt2img_grids": OptionInfo("outputs/txt2img-grids", 'Output directory for txt2img grids', component_args=hide_dirs),

     "outdir_img2img_grids": OptionInfo("outputs/img2img-grids", 'Output directory for img2img grids', component_args=hide_dirs),

     "outdir_save": OptionInfo("log/images", "Directory for saving images using the Save button", component_args=hide_dirs),

+    "outdir_init_images": OptionInfo("outputs/init-images", "Directory for saving init images when using img2img", component_args=hide_dirs),

 }))

 

 options_templates.update(options_section(('saving-to-dirs', "Saving to a directory"), {