diff options
author | timntorres <timothynarcisotorres@gmail.com> | 2022-12-17 14:28:51 +0000 |
---|---|---|
committer | timntorres <timothynarcisotorres@gmail.com> | 2022-12-17 16:50:20 +0000 |
commit | a7a039d53a69f8c32cb889fe322e769b238fec27 (patch) | |
tree | e2d562385b939a5251644eccf700fee38156f842 | |
parent | a26fe85056cf0dacef2d78cccf6ab100fd16da1c (diff) | |
download | stable-diffusion-webui-gfx803-a7a039d53a69f8c32cb889fe322e769b238fec27.tar.gz stable-diffusion-webui-gfx803-a7a039d53a69f8c32cb889fe322e769b238fec27.tar.bz2 stable-diffusion-webui-gfx803-a7a039d53a69f8c32cb889fe322e769b238fec27.zip |
Add option to include upscaler name in filename.
-rw-r--r-- | modules/extras.py | 5 | ||||
-rw-r--r-- | modules/shared.py | 1 |
2 files changed, 4 insertions, 2 deletions
diff --git a/modules/extras.py b/modules/extras.py index 9b60e360..074a7c22 100644 --- a/modules/extras.py +++ b/modules/extras.py @@ -194,8 +194,9 @@ def run_extras(extras_mode, resize_mode, image, image_folder, input_dir, output_ 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_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,
diff --git a/modules/shared.py b/modules/shared.py index dc45fcaa..218894e8 100644 --- a/modules/shared.py +++ b/modules/shared.py @@ -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, "Add upscaler name to the end of filename 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"),
|