mirrors/stable-diffusion-webui
1
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2026-01-30 04:11:15 -08:00
Code Issues Projects Releases Packages Wiki Activity

Add types to split_grid

Browse source
This commit is contained in:
Aarni Koskela 2023-12-25 14:44:15 +02:00
parent 31992eff9b
commit 7aa27b000a
1 changed files with 2 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

5
modules/images.py
View file

@ -64,9 +64,8 @@ def image_grid(imgs, batch_size=1, rows=None):
Grid = namedtuple("Grid", ["tiles", "tile_w", "tile_h", "image_w", "image_h", "overlap"])
def split_grid(image, tile_w=512, tile_h=512, overlap=64):
w = image.width
h = image.height
def split_grid(image: Image.Image, tile_w: int = 512, tile_h: int = 512, overlap: int = 64) -> Grid:
w, h = image.size
non_overlap_width = tile_w - overlap
non_overlap_height = tile_h - overlap