diff options
| author | Aarni Koskela <akx@iki.fi> | 2023-07-03 10:08:28 +0000 |
|---|---|---|
| committer | Aarni Koskela <akx@iki.fi> | 2023-07-03 10:08:28 +0000 |
| commit | 08f9b705cda4277aed49ed00c405ada2925e3b50 (patch) | |
| tree | 60f765041ecc4e71fbccf76e221aecc3cd1a7ad5 | |
| parent | fab73f2e7d388ca99cdb3d5de7f36c0b9a1a3b1c (diff) | |
| download | stable-diffusion-webui-gfx803-08f9b705cda4277aed49ed00c405ada2925e3b50.tar.gz stable-diffusion-webui-gfx803-08f9b705cda4277aed49ed00c405ada2925e3b50.tar.bz2 stable-diffusion-webui-gfx803-08f9b705cda4277aed49ed00c405ada2925e3b50.zip | |
Use read_info_from_image in postprocessing
Avoids bad keys such as `exif` ending up in the "PNG info" passed forward
| -rw-r--r-- | modules/postprocessing.py | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/modules/postprocessing.py b/modules/postprocessing.py index 736315e2..38544c38 100644 --- a/modules/postprocessing.py +++ b/modules/postprocessing.py @@ -54,7 +54,9 @@ def run_postprocessing(extras_mode, image, image_folder, input_dir, output_dir, for image, name in zip(image_data, image_names):
shared.state.textinfo = name
- existing_pnginfo = image.info or {}
+ parameters, existing_pnginfo = images.read_info_from_image(image)
+ if parameters:
+ existing_pnginfo["parameters"] = parameters
pp = scripts_postprocessing.PostprocessedImage(image.convert("RGB"))
|