apply lint
This commit is contained in:
parent
440e47f736
commit
3b6eb6c43b
|
@ -279,9 +279,20 @@ def get_tile_latents(
|
||||||
xt = x + t
|
xt = x + t
|
||||||
yt = y + t
|
yt = y + t
|
||||||
|
|
||||||
|
logger.trace(
|
||||||
|
"getting tile latents: [%s:%s, %s:%s] within %s",
|
||||||
|
y,
|
||||||
|
yt,
|
||||||
|
x,
|
||||||
|
xt,
|
||||||
|
full_latents.shape,
|
||||||
|
)
|
||||||
|
|
||||||
tile_latents = full_latents[:, :, y:yt, x:xt]
|
tile_latents = full_latents[:, :, y:yt, x:xt]
|
||||||
|
|
||||||
if tile_latents.shape != full_latents.shape and (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 = get_latents_from_seed(seed, size, batch=tile_latents.shape[0])
|
||||||
extra_latents[
|
extra_latents[
|
||||||
:, :, 0 : tile_latents.shape[2], 0 : tile_latents.shape[3]
|
:, :, 0 : tile_latents.shape[2], 0 : tile_latents.shape[3]
|
||||||
|
|
Loading…
Reference in New Issue