fix(api): avoid padding latents when they are not being tiled
This commit is contained in:
parent
944c92b824
commit
d94a8d6f6e
|
@ -281,7 +281,7 @@ def get_tile_latents(
|
|||
|
||||
tile_latents = full_latents[:, :, y:yt, x:xt]
|
||||
|
||||
if tile_latents.shape[2] < t or tile_latents.shape[3] < t:
|
||||
if tile_latents.shape != full_latents.shape and (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 : tile_latents.shape[2], 0 : tile_latents.shape[3]
|
||||
|
|
Loading…
Reference in New Issue