Skip to content

H
Hydra Node Http
Commit c7f0a0ac authored by gd1551's avatar gd1551 Committed by GitHub
Browse files
Options

oops

parent 4aca652b
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
main.py
View file @ c7f0a0ac
...@@ -103,6 +103,7 @@ def generate(request: GenerationRequest): ...@@ -103,6 +103,7 @@ def generate(request: GenerationRequest):
else: else:
images = model.sample(request) images = model.sample(request)
prompt = request.prompt
del request.prompt del request.prompt
if request.image is not None: if request.image is not None:
del request.image del request.image
...@@ -112,7 +113,7 @@ def generate(request: GenerationRequest): ...@@ -112,7 +113,7 @@ def generate(request: GenerationRequest):
for x in range(len(images)): for x in range(len(images)):
metadata = PngInfo() metadata = PngInfo()
metadata.add_text("Title", "AI generated image") metadata.add_text("Title", "AI generated image")
metadata.add_text("Description", request.prompt) metadata.add_text("Description", prompt)
metadata.add_text("Software", "NovelAI") metadata.add_text("Software", "NovelAI")
metadata.add_text("Source", "Stable Diffusion "+model_hash) metadata.add_text("Source", "Stable Diffusion "+model_hash)
if seed is not None: if seed is not None:
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment