wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-02-09 21:12:36 +08:00
Code Issues Actions 12 Packages Projects Releases Wiki Activity

Fix extra chunk metadata add

Browse Source
This commit is contained in:
catboxanon 2025-04-14 12:52:52 -04:00
parent 5d6bca8e78
commit 8060017fe0
1 changed files with 1 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
nodes.py
View File

@ -1627,10 +1627,7 @@ class SaveImage:
metadata.add_text(x, json.dumps(extra_pnginfo[x])) metadata.add_text(x, json.dumps(extra_pnginfo[x]))
if hasattr(images, "png_chunks"): if hasattr(images, "png_chunks"):
for name, data in images.png_chunks.items(): for name, data in images.png_chunks.items():
if name in self.extra_chunks: metadata.add(name, data)
metadata.add(name.lower(), data)
else:
metadata.add(name, data)
filename_with_batch_num = filename.replace("%batch_num%", str(batch_number)) filename_with_batch_num = filename.replace("%batch_num%", str(batch_number))
file = f"{filename_with_batch_num}_{counter:05}_.png" file = f"{filename_with_batch_num}_{counter:05}_.png"
img.save(os.path.join(full_output_folder, file), pnginfo=metadata, compress_level=self.compress_level) img.save(os.path.join(full_output_folder, file), pnginfo=metadata, compress_level=self.compress_level)