diff --git a/api/onnx_web/diffusers/utils.py b/api/onnx_web/diffusers/utils.py index 4756bee4..b66fafbb 100644 --- a/api/onnx_web/diffusers/utils.py +++ b/api/onnx_web/diffusers/utils.py @@ -281,7 +281,7 @@ def get_tile_latents( tile_latents = full_latents[:, :, y:yt, x:xt] - if tile_latents.shape[2] < t or tile_latents.shape[3] < t: + if tile_latents.shape != full_latents.shape and (tile_latents.shape[2] < t or tile_latents.shape[3] < t): extra_latents = get_latents_from_seed(seed, size, batch=tile_latents.shape[0]) extra_latents[ :, :, 0 : tile_latents.shape[2], 0 : tile_latents.shape[3]