Revert extra return statement

This commit is contained in:
awsr
2026-03-24 06:28:39 -07:00
parent a6f6a37dea
commit fd1646cc37
-1
View File
@@ -54,7 +54,6 @@ def get_grid_size(imgs, batch_size=1, rows: int | None = None, cols: int | None
while len(imgs) % rows != 0:
rows -= 1
cols = math.ceil(len(imgs) / rows)
return rows, cols
elif rows is not None and cols is None:
cols = math.ceil(len(imgs) / rows)
elif rows is None and cols is not None: