From 5ad7b4c1b4fbc6b958384cd8c2a1c45a98816db2 Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Tue, 2 Jan 2024 21:19:34 -0600 Subject: [PATCH] fix thumbnail call --- api/onnx_web/diffusers/run.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/api/onnx_web/diffusers/run.py b/api/onnx_web/diffusers/run.py index 9e902cc2..eb50b76b 100644 --- a/api/onnx_web/diffusers/run.py +++ b/api/onnx_web/diffusers/run.py @@ -123,9 +123,11 @@ def run_txt2img_pipeline( # add a thumbnail, if requested cover = images[0] if params.thumbnail and (cover.width > 512 or cover.height > 512): - thumbnail = cover.thumbnail((512, 512)) - images.insert(thumbnail) - outputs.insert(f"{worker.name}-thumb.{server.image_format}") + thumbnail = cover.copy() + thumbnail.thumbnail((512, 512)) + + images.insert(0, thumbnail) + outputs.insert(0, f"{worker.name}-thumb.{server.image_format}") for image, output in zip(images, outputs): logger.trace("saving output image %s: %s", output, image.size)