diff options
| author | AUTOMATIC <16777216c@gmail.com> | 2022-10-09 10:10:15 +0000 |
|---|---|---|
| committer | AUTOMATIC <16777216c@gmail.com> | 2022-10-09 10:10:15 +0000 |
| commit | bd833409ac7b8337040d521f6b65ced51e1b2ea8 (patch) | |
| tree | 9958b09f09d94645c80e4ad693d036054892cdd7 /modules/extras.py | |
| parent | 0609ce06c0778536cb368ac3867292f87c6d9fc7 (diff) | |
| download | stable-diffusion-webui-gfx803-bd833409ac7b8337040d521f6b65ced51e1b2ea8.tar.gz stable-diffusion-webui-gfx803-bd833409ac7b8337040d521f6b65ced51e1b2ea8.tar.bz2 stable-diffusion-webui-gfx803-bd833409ac7b8337040d521f6b65ced51e1b2ea8.zip | |
additional changes for saving pnginfo for #1803
Diffstat (limited to 'modules/extras.py')
| -rw-r--r-- | modules/extras.py | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/modules/extras.py b/modules/extras.py index ef6e6de7..39dd3806 100644 --- a/modules/extras.py +++ b/modules/extras.py @@ -98,6 +98,10 @@ def run_extras(extras_mode, image, image_folder, gfpgan_visibility, codeformer_v no_prompt=True, grid=False, pnginfo_section_name="extras", existing_info=existing_pnginfo,
forced_filename=image_name if opts.use_original_name_batch else None)
+ if opts.enable_pnginfo:
+ image.info = existing_pnginfo
+ image.info["extras"] = info
+
outputs.append(image)
devices.torch_gc()
|