1
0
Fork 0

fix(api): match size of existing latent patch

This commit is contained in:
Sean Sube 2023-07-12 22:02:05 -05:00
parent 3d4c77d5d0
commit cf95ad2f0e
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 1 additions and 1 deletions

View File

@ -286,7 +286,7 @@ def get_tile_latents(
if 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:t, 0:t] = tile_latents
extra_latents[:, :, 0:tile_latents.shape[2], 0:tile_latents.shape[3]] = tile_latents
tile_latents = extra_latents
return tile_latents