Commit a26fe850 authored by timntorres's avatar timntorres
Browse files

Add upscaler name as a suffix.

parent 44c46f0e
Loading
Loading
Loading
Loading
+6 −1
Original line number Diff line number Diff line
@@ -193,8 +193,13 @@ 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 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