Add option to save before color correction. Add suffix param when saving files, used for special saves without color correction and face restoration.

1 files changed, 2 insertions, 2 deletions

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()