apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

apply lint

This commit is contained in:
Sean Sube 2023-08-20 15:18:47 -05:00
parent 440e47f736
commit 3b6eb6c43b
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 12 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
api/onnx_web/diffusers/utils.py
View File

@ -279,9 +279,20 @@ def get_tile_latents(
xt = x + 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]
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[
:, :, 0 : tile_latents.shape[2], 0 : tile_latents.shape[3]