Unverified Commit 55f3ef87 authored by AUTOMATIC1111's avatar AUTOMATIC1111 Committed by GitHub
Browse files

Merge pull request #5814 from timntorres/5802-save-upscaler-to-filename 

Add option to save upscaler to filename suffix in extras tab.
parents 992a877a 6fd91c91

modules/extras.py

+7 −1
Original line number Diff line number Diff line
@@ -193,8 +193,14 @@ def run_extras(extras_mode, resize_mode, image, image_folder, input_dir, output_ 
        else:

            basename = ''



        # Add upscaler name as a suffix.

        suffix = f"-{shared.sd_upscalers[extras_upscaler_1].name}" if shared.opts.use_upscaler_name_as_suffix else ""

        # Add second upscaler if applicable.

        if suffix and extras_upscaler_2 and extras_upscaler_2_visibility:

            suffix += f"-{shared.sd_upscalers[extras_upscaler_2].name}"



        images.save_image(image, path=outpath, basename=basename, seed=None, prompt=None, extension=opts.samples_format, info=info, short_filename=True,

                          no_prompt=True, grid=False, pnginfo_section_name="extras", existing_info=existing_pnginfo, forced_filename=None)

                          no_prompt=True, grid=False, pnginfo_section_name="extras", existing_info=existing_pnginfo, forced_filename=None, suffix=suffix)



        if opts.enable_pnginfo:

            image.info = existing_pnginfo

modules/shared.py

+1 −0
Original line number Diff line number Diff line
@@ -293,6 +293,7 @@ options_templates.update(options_section(('saving-images', "Saving images/grids" 
    "export_for_4chan": OptionInfo(True, "If PNG image is larger than 4MB or any dimension is larger than 4000, downscale and save copy as JPG"),



    "use_original_name_batch": OptionInfo(False, "Use original name for output filename during batch process in extras tab"),

    "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"),

CRA Git | Maintained and supported by SUSTech CRA and CCSE