From 25eb1e3d9072a060a253abce6889e7fef440dfe1 Mon Sep 17 00:00:00 2001 From: Robin Fernandes Date: Thu, 22 Sep 2022 20:54:50 +1000 Subject: Add option to save before color correction. Add suffix param when saving files, used for special saves without color correction and face restoration. --- modules/images.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'modules/images.py') diff --git a/modules/images.py b/modules/images.py index d1707263..f65f4553 100644 --- a/modules/images.py +++ b/modules/images.py @@ -311,7 +311,7 @@ def get_next_sequence_number(path, basename): return result + 1 -def save_image(image, path, basename, seed=None, prompt=None, extension='png', info=None, short_filename=False, no_prompt=False, grid=False, pnginfo_section_name='parameters', p=None, existing_info=None, forced_filename=None): +def save_image(image, path, basename, seed=None, prompt=None, extension='png', info=None, short_filename=False, no_prompt=False, grid=False, pnginfo_section_name='parameters', p=None, existing_info=None, forced_filename=None, suffix=""): if short_filename or prompt is None or seed is None: file_decoration = "" elif opts.save_to_dirs: @@ -322,7 +322,7 @@ def save_image(image, path, basename, seed=None, prompt=None, extension='png', i if file_decoration != "": file_decoration = "-" + file_decoration.lower() - file_decoration = apply_filename_pattern(file_decoration, p, seed, prompt) + file_decoration = apply_filename_pattern(file_decoration, p, seed, prompt) + suffix if extension == 'png' and opts.enable_pnginfo and info is not None: pnginfo = PngImagePlugin.PngInfo() -- cgit v1.2.1