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 missing seed in filename

Browse Source
This commit is contained in:
psychedelicious 2022-09-21 00:26:05 +10:00
parent cae0579ba9
commit 3e8a861fc0
1 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
backend/server.py
View File

@ -36,7 +36,10 @@ output_dir = "outputs/" # Base output directory for images
# host = 'localhost' # Web & socket.io host # host = 'localhost' # Web & socket.io host
host = "localhost" # Web & socket.io host host = "localhost" # Web & socket.io host
port = 9090 # Web & socket.io port port = 9090 # Web & socket.io port
verbose = False # enables copious socket.io logging
additional_allowed_origins = [
"http://localhost:5173"
] # additional CORS allowed origins
model = "stable-diffusion-1.4" model = "stable-diffusion-1.4"
""" """
@ -502,12 +505,10 @@ def write_log_message(message, log_path=log_path):
def save_image( def save_image(
image, command, metadata, output_dir, step_index=None, postprocessing=False image, command, metadata, output_dir, step_index=None, postprocessing=False
): ):
seed = metadata["seed"] if "seed" in metadata else "unknown_seed"
pngwriter = PngWriter(output_dir) pngwriter = PngWriter(output_dir)
prefix = pngwriter.unique_prefix() prefix = pngwriter.unique_prefix()
filename = f"{prefix}.{seed}" filename = f"{prefix}.{metadata['image']['seed']}"
if step_index: if step_index:
filename += f".{step_index}" filename += f".{step_index}"