OPENGIT/stable-diffusion-webui
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2025-04-24 22:09:00 +08:00
Code Issues Packages Projects Releases 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
Show Stats 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"]) 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): def split_grid(image: Image.Image, tile_w: int = 512, tile_h: int = 512, overlap: int = 64) -> Grid:
w = image.width w, h = image.size
h = image.height
non_overlap_width = tile_w - overlap non_overlap_width = tile_w - overlap
non_overlap_height = tile_h - overlap non_overlap_height = tile_h - overlap