github/InvokeAI
1
1
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI.git synced 2026-04-23 03:00:31 -04:00
Code Issues Packages Projects Releases Wiki Activity

feat(app): embed PNG info in invokeai_metadata and invokeai_graph

Browse Source 
Using just `metadata` and `graph` feel a bit too generic.
This commit is contained in:
psychedelicious
2023-07-13 12:49:10 +10:00
parent 6bea7bac36
commit 5f5c93abb4
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
invokeai/app/services/image_file_storage.py
View File

@@ -122,9 +122,9 @@ class DiskImageFileStorage(ImageFileStorageBase):
pnginfo = PngImagePlugin.PngInfo()
if metadata is not None:
pnginfo.add_text("metadata", json.dumps(metadata))
pnginfo.add_text("invokeai_metadata", json.dumps(metadata))
if graph is not None:
pnginfo.add_text("graph", json.dumps(graph))
pnginfo.add_text("invokeai_graph", json.dumps(graph))
image.save(image_path, "PNG", pnginfo=pnginfo)
thumbnail_name = get_thumbnail_name(image_name)