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

add sd-metadata metadata_loads() and metadata_dumps() functions

Browse Source
This commit is contained in:
Lincoln Stein
2022-09-17 13:28:37 -04:00
parent 2faa116238
commit 239f41f3e0
3 changed files with 68 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
scripts/dream.py
View File

@ -8,7 +8,7 @@ import copy
import warnings
import time
import ldm.dream.readline
from ldm.dream.args import Args, format_metadata
from ldm.dream.args import Args, metadata_dumps
from ldm.dream.pngwriter import PngWriter
from ldm.dream.server import DreamServer, ThreadingDreamServer
from ldm.dream.image_util import make_grid
@ -245,10 +245,9 @@ def main_loop(gen, opt, infile):
path = file_writer.save_image_and_prompt_to_png(
image = image,
dream_prompt = formatted_dream_prompt,
metadata = format_metadata(
metadata = metadata_dumps(
opt,
seeds = [seed],
weights = gen.weights,
model_hash = gen.model_hash,
),
name = filename,
@ -272,7 +271,7 @@ def main_loop(gen, opt, infile):
filename = f'{prefix}.{first_seed}.png'
formatted_dream_prompt = opt.dream_prompt_str(seed=first_seed,grid=True,iterations=len(grid_images))
formatted_dream_prompt += f' # {grid_seeds}'
metadata = format_metadata(
metadata = metadata.dumps(
opt,
seeds = grid_seeds,
weights = gen.weights,