From 540864c3a8ae62d7c40d2641a777119f3ed1fee6 Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Mon, 4 Sep 2023 21:18:07 -0500 Subject: [PATCH] fix initial tile size for XL --- api/onnx_web/chain/source_txt2img.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/api/onnx_web/chain/source_txt2img.py b/api/onnx_web/chain/source_txt2img.py index 75c9e78a..cc642d55 100644 --- a/api/onnx_web/chain/source_txt2img.py +++ b/api/onnx_web/chain/source_txt2img.py @@ -28,7 +28,7 @@ class SourceTxt2ImgStage(BaseStage): self, worker: WorkerContext, server: ServerContext, - _stage: StageParams, + stage: StageParams, params: ImageParams, _source: Image.Image, *, @@ -59,7 +59,10 @@ class SourceTxt2ImgStage(BaseStage): params ) - tile_size = params.tiles + if params.is_xl(): + tile_size = max(stage.tile_size, params.tiles) + else: + tile_size = params.tiles # this works for panorama as well, because tile_size is already max(tile_size, *size) latent_size = size.min(tile_size, tile_size)