apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

generate a thumbnail when the first image is > 512

This commit is contained in:
Sean Sube 2024-01-02 20:58:02 -06:00
parent b35101ebbd
commit 5de28d0361
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
2 changed files with 12 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
api/onnx_web/diffusers/run.py
View File

@ -120,6 +120,13 @@ def run_txt2img_pipeline(
_pairs, loras, inversions, _rest = parse_prompt(params) _pairs, loras, inversions, _rest = parse_prompt(params)
# 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}")
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)
dest = save_image( dest = save_image(

5
api/onnx_web/params.py
View File

@ -213,6 +213,7 @@ class ImageParams:
vae_tile: int vae_tile: int
vae_overlap: float vae_overlap: float
denoise: int denoise: int
thumbnail: int
def __init__( def __init__(
self, self,
@ -236,6 +237,7 @@ class ImageParams:
vae_overlap: float = 0.25, vae_overlap: float = 0.25,
vae_tile: int = 512, vae_tile: int = 512,
denoise: int = 3, denoise: int = 3,
thumbnail: int = 1,
) -> None: ) -> None:
self.model = model self.model = model
self.pipeline = pipeline self.pipeline = pipeline
@ -257,6 +259,7 @@ class ImageParams:
self.vae_overlap = vae_overlap self.vae_overlap = vae_overlap
self.vae_tile = vae_tile self.vae_tile = vae_tile
self.denoise = denoise self.denoise = denoise
self.thumbnail = thumbnail
def do_cfg(self): def do_cfg(self):
return self.cfg > 1.0 return self.cfg > 1.0
@ -328,6 +331,7 @@ class ImageParams:
"vae_overlap": self.vae_overlap, "vae_overlap": self.vae_overlap,
"vae_tile": self.vae_tile, "vae_tile": self.vae_tile,
"denoise": self.denoise, "denoise": self.denoise,
"thumbnail": self.thumbnail,
} }
def with_args(self, **kwargs): def with_args(self, **kwargs):
@ -352,6 +356,7 @@ class ImageParams:
kwargs.get("vae_overlap", self.vae_overlap), kwargs.get("vae_overlap", self.vae_overlap),
kwargs.get("vae_tile", self.vae_tile), kwargs.get("vae_tile", self.vae_tile),
kwargs.get("denoise", self.denoise), kwargs.get("denoise", self.denoise),
kwargs.get("thumbnail", self.thumbnail),
) )