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( def resize_latent_shape(
latents: np.ndarray, latents: np.ndarray,
size: Size, size: Tuple[int, int],
) -> Tuple[int, int, 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( def get_tile_latents(