apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix latent resizing

This commit is contained in:
Sean Sube 2023-12-03 11:48:52 -06:00
parent d759c31bb6
commit 05b1ef611f
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
api/onnx_web/diffusers/utils.py
View File

@ -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(