Fix assertion when using tiling with padding=blending=0

This commit is contained in:
Acly
2025-01-26 11:57:11 +01:00
parent 8ed5591574
commit bed8b36705
+1 -1
View File
@@ -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