From 05b1ef611f3060ae6957fb988b05174c9b784179 Mon Sep 17 00:00:00 2001
From: Sean Sube <seansube@gmail.com>
Date: Sun, 3 Dec 2023 11:48:52 -0600
Subject: [PATCH] fix latent resizing

---
 api/onnx_web/diffusers/utils.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/api/onnx_web/diffusers/utils.py b/api/onnx_web/diffusers/utils.py
index 19f3130d..ff39d371 100644
--- a/api/onnx_web/diffusers/utils.py
+++ b/api/onnx_web/diffusers/utils.py
@@ -296,9 +296,9 @@ def expand_latents(
 
 def resize_latent_shape(
     latents: np.ndarray,
-    size: Size,
+    size: Tuple[int, int],
 ) -> Tuple[int, int, int, int]:
-    return (latents.shape[0], latents.shape[1], size.height, size.width)
+    return (latents.shape[0], latents.shape[1], *size)
 
 
 def get_tile_latents(