aboutsummaryrefslogtreecommitdiff
path: root/modules/extras.py
diff options
context:
space:
mode:
authortimntorres <timothynarcisotorres@gmail.com>2022-12-17 04:31:03 -0800
committertimntorres <timothynarcisotorres@gmail.com>2022-12-17 05:11:06 -0800
commita26fe85056cf0dacef2d78cccf6ab100fd16da1c (patch)
tree335e7da432a59f98a3989a78b80427218d544e6c /modules/extras.py
parent44c46f0ed395967cd3830dd481a2db759fda5b3b (diff)
Add upscaler name as a suffix.
Diffstat (limited to 'modules/extras.py')
-rw-r--r--modules/extras.py7
1 files changed, 6 insertions, 1 deletions
diff --git a/modules/extras.py b/modules/extras.py
index bc349d5e..9b60e360 100644
--- a/modules/extras.py
+++ b/modules/extras.py
@@ -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