From 3b6eb6c43bc86b36f30ffc09b9facd3abaaf6d04 Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Sun, 20 Aug 2023 15:18:47 -0500 Subject: [PATCH] apply lint --- api/onnx_web/diffusers/utils.py | 13 ++++++++++++- 1 file changed, 12 insertions(+), 1 deletion(-) diff --git a/api/onnx_web/diffusers/utils.py b/api/onnx_web/diffusers/utils.py index b66fafbb..0c4ade2c 100644 --- a/api/onnx_web/diffusers/utils.py +++ b/api/onnx_web/diffusers/utils.py @@ -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]