fix(api): avoid padding latents when they are not being tiled

ssube · Sep 5, 2023 · d94a8d6 · d94a8d6
1 parent 944c92b
commit d94a8d6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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]