prevent latent tiles from going out of bounds

2023-07-09 22:35:15 -05:00 · 2023-07-09 22:35:15 -05:00 · da6f00d2a5
parent 76bed9582b
commit da6f00d2a5
1 changed files with 2 additions and 2 deletions
--- a/api/onnx_web/diffusers/utils.py
+++ b/api/onnx_web/diffusers/utils.py
@ -273,8 +273,8 @@ def get_tile_latents(
 ) -> np.ndarray:
    x, y, tile = dims
    t = tile // LATENT_FACTOR
-    x = x // LATENT_FACTOR
-    y = y // LATENT_FACTOR
+    x = max(0, x // LATENT_FACTOR)
+    y = max(0, y // LATENT_FACTOR)
    xt = x + t
    yt = y + t