fix thumbnail call
This commit is contained in:
parent
5de28d0361
commit
5ad7b4c1b4
|
@ -123,9 +123,11 @@ def run_txt2img_pipeline(
|
||||||
# add a thumbnail, if requested
|
# add a thumbnail, if requested
|
||||||
cover = images[0]
|
cover = images[0]
|
||||||
if params.thumbnail and (cover.width > 512 or cover.height > 512):
|
if params.thumbnail and (cover.width > 512 or cover.height > 512):
|
||||||
thumbnail = cover.thumbnail((512, 512))
|
thumbnail = cover.copy()
|
||||||
images.insert(thumbnail)
|
thumbnail.thumbnail((512, 512))
|
||||||
outputs.insert(f"{worker.name}-thumb.{server.image_format}")
|
|
||||||
|
images.insert(0, thumbnail)
|
||||||
|
outputs.insert(0, f"{worker.name}-thumb.{server.image_format}")
|
||||||
|
|
||||||
for image, output in zip(images, outputs):
|
for image, output in zip(images, outputs):
|
||||||
logger.trace("saving output image %s: %s", output, image.size)
|
logger.trace("saving output image %s: %s", output, image.size)
|
||||||
|
|
Loading…
Reference in New Issue