From bd833409ac7b8337040d521f6b65ced51e1b2ea8 Mon Sep 17 00:00:00 2001
From: AUTOMATIC <16777216c@gmail.com>
Date: Sun, 9 Oct 2022 13:10:15 +0300
Subject: additional changes for saving pnginfo for #1803

---
 modules/extras.py | 4 ++++
 1 file changed, 4 insertions(+)

(limited to 'modules/extras.py')

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()
-- 
cgit v1.2.1