Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00: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
Show Stats 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)