aboutsummaryrefslogtreecommitdiffstats
path: root/modules/images.py
diff options
context:
space:
mode:
authorAUTOMATIC <16777216c@gmail.com>2022-09-11 08:31:16 +0000
committerAUTOMATIC <16777216c@gmail.com>2022-09-11 08:31:16 +0000
commitf1858744189ad54bb464c97b9c735275795a6f53 (patch)
treebbe4d85f56aa4d2ef262aada3f06641477bf0faf /modules/images.py
parent2e6153e343e5002b5c161d00a43453400be2d884 (diff)
downloadstable-diffusion-webui-gfx803-f1858744189ad54bb464c97b9c735275795a6f53.tar.gz
stable-diffusion-webui-gfx803-f1858744189ad54bb464c97b9c735275795a6f53.tar.bz2
stable-diffusion-webui-gfx803-f1858744189ad54bb464c97b9c735275795a6f53.zip
[Feature Request] Save defaults for extras & keep image parameters after using extras #251
Diffstat (limited to 'modules/images.py')
-rw-r--r--modules/images.py4
1 files changed, 2 insertions, 2 deletions
diff --git a/modules/images.py b/modules/images.py
index c1a58013..26c399b6 100644
--- a/modules/images.py
+++ b/modules/images.py
@@ -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