hoelee/stable-diffusion-webui
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2025-08-08 05:12:35 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

[Feature Request] Save defaults for extras & keep image parameters after using extras #251

Browse Source
This commit is contained in:
AUTOMATIC
2022-09-11 11:31:16 +03:00
parent 2e6153e343
commit f185874418
4 changed files with 94 additions and 83 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
modules/images.py
View File

@@ -243,7 +243,7 @@ def sanitize_filename_part(text):
return text.replace(' ', '_').translate({ord(x): '' for x in invalid_filename_chars})[:128]
def save_image(image, path, basename, seed=None, prompt=None, extension='png', info=None, short_filename=False, no_prompt=False):
def save_image(image, path, basename, seed=None, prompt=None, extension='png', info=None, short_filename=False, no_prompt=False, pnginfo_section_name='parameters'):
# would be better to add this as an argument in future, but will do for now
is_a_grid = basename != ""
@@ -256,7 +256,7 @@ def save_image(image, path, basename, seed=None, prompt=None, extension='png', i
if extension == 'png' and opts.enable_pnginfo and info is not None:
pnginfo = PngImagePlugin.PngInfo()
pnginfo.add_text("parameters", info)
pnginfo.add_text(pnginfo_section_name, info)
else:
pnginfo = None