additional changes for saving pnginfo for #1803

2025-08-04 03:10:21 +00:00 · 2022-10-09 13:10:15 +03:00
parent 0609ce06c0
commit bd833409ac
2 changed files with 8 additions and 2 deletions
--- 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()