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

correct bug when trying to enhance JPG images (#1928)

Browse Source 
This fix was authored by @mebelz and is reissued here to base it on
`main`.
This commit is contained in:
Lincoln Stein 2022-12-11 13:48:47 -05:00 committed by GitHub
parent 7efe0f3996
commit f745f78cb3
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
ldm/invoke/pngwriter.py
View File

@ -57,8 +57,13 @@ def retrieve_metadata(img_path):
metadata stored there, as a dict metadata stored there, as a dict
''' '''
im = Image.open(img_path) im = Image.open(img_path)
md = im.text.get('sd-metadata', '{}') if hasattr(im, 'text'):
dream_prompt = im.text.get('Dream', '') md = im.text.get('sd-metadata', '{}')
dream_prompt = im.text.get('Dream', '')
else:
# When trying to retrieve metadata from images without a 'text' payload, such as JPG images.
md = '{}'
dream_prompt = ''
return {'sd-metadata': json.loads(md), 'Dream': dream_prompt} return {'sd-metadata': json.loads(md), 'Dream': dream_prompt}
def write_metadata(img_path:str, meta:dict): def write_metadata(img_path:str, meta:dict):