Commit 4aca652b authored by gd1551's avatar gd1551 Committed by GitHub

don't copy the same request object for each image as it's not used after generation

parent d9a76dd0
...@@ -103,6 +103,10 @@ def generate(request: GenerationRequest): ...@@ -103,6 +103,10 @@ def generate(request: GenerationRequest):
else: else:
images = model.sample(request) images = model.sample(request)
del request.prompt
if request.image is not None:
del request.image
seed = request.seed seed = request.seed
images_encoded = [] images_encoded = []
for x in range(len(images)): for x in range(len(images)):
...@@ -111,14 +115,11 @@ def generate(request: GenerationRequest): ...@@ -111,14 +115,11 @@ def generate(request: GenerationRequest):
metadata.add_text("Description", request.prompt) metadata.add_text("Description", request.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)
request_copy = request.copy()
del request_copy.prompt
if request_copy.image is not None:
del request_copy.image
if seed is not None: if seed is not None:
request_copy.seed = seed request.seed = seed
seed += 1 seed += 1
metadata.add_text("Comment", json.dumps(request_copy)) print(request, json.dumps(request))
metadata.add_text("Comment", json.dumps(request))
#image = simplejpeg.encode_jpeg(images[x], quality=95) #image = simplejpeg.encode_jpeg(images[x], quality=95)
image = Image.fromarray(images[x]) image = Image.fromarray(images[x])
#save pillow image with bytesIO #save pillow image with bytesIO
......
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