generate a thumbnail when the first image is > 512
This commit is contained in:
parent
b35101ebbd
commit
5de28d0361
|
@ -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(
|
||||||
|
|
|
@ -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),
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue