fix(api): match size of existing latent patch
This commit is contained in:
parent
3d4c77d5d0
commit
cf95ad2f0e
|
@ -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
|
||||
|
|
Loading…
Reference in New Issue