aboutsummaryrefslogtreecommitdiff
path: root/modules/extras.py
diff options
context:
space:
mode:
authortimntorres <timothynarcisotorres@gmail.com>2022-12-17 06:28:51 -0800
committertimntorres <timothynarcisotorres@gmail.com>2022-12-17 08:50:20 -0800
commita7a039d53a69f8c32cb889fe322e769b238fec27 (patch)
treee2d562385b939a5251644eccf700fee38156f842 /modules/extras.py
parenta26fe85056cf0dacef2d78cccf6ab100fd16da1c (diff)
Add option to include upscaler name in filename.
Diffstat (limited to 'modules/extras.py')
-rw-r--r--modules/extras.py5
1 files changed, 3 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,