From bed8b36705a213f2ec452f8f090b2bcdf43f19ac Mon Sep 17 00:00:00 2001 From: Acly Date: Sun, 26 Jan 2025 11:57:11 +0100 Subject: [PATCH] Fix assertion when using tiling with padding=blending=0 --- tile.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tile.py b/tile.py index 2d14d9a..0844e25 100644 --- a/tile.py +++ b/tile.py @@ -36,7 +36,7 @@ class TileLayout: def init(self, image: Tensor, min_tile_size: int, padding: int, blending: int): assert all([x % 8 == 0 for x in image.shape[-3:-1]]), "Image size must be divisible by 8" assert min_tile_size % 8 == 0, "Tile size must be divisible by 8" - assert blending < padding, "Blending must be smaller than padding" + assert blending <= padding, "Blending must be smaller than padding" self.image_size = np.array(image.shape[-3:-1]) self.padding = padding