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

Fixes metadata_loads() #686

Browse Source
This commit is contained in:
psychedelicious 2022-09-19 11:53:47 +10:00 committed by Lincoln Stein
parent 6bf19cd897
commit 682d74754c
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
ldm/dream/args.py
View File

@ -681,7 +681,10 @@ def metadata_loads(metadata):
''' '''
results = [] results = []
try: try:
images = metadata['sd-metadata']['images'] if 'grid' in metadata['sd-metadata']:
images = metadata['sd-metadata']['images']
else:
images = [metadata['sd-metadata']['image']]
for image in images: for image in images:
# repack the prompt and variations # repack the prompt and variations
image['prompt'] = ','.join([':'.join([x['prompt'], str(x['weight'])]) for x in image['prompt']]) image['prompt'] = ','.join([':'.join([x['prompt'], str(x['weight'])]) for x in image['prompt']])