RomanNum3ral
/
ai-toolkit
mirror of https://github.com/ostris/ai-toolkit.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Round buckets for divisibility instead of always rounding down.

This commit is contained in:
Jaret Burkett 2026-05-28 09:08:21 -06:00
parent c9264bdd0b
commit 15082cfb8a
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
toolkit/buckets.py
View File

@ -22,7 +22,7 @@ def get_bucket_for_image_size(
target_pixels = min(total_pixels, max_pixels)
scaler = target_pixels / total_pixels
new_width = int((width * scaler) // divisibility * divisibility)
new_height = int((height * scaler) // divisibility * divisibility)
new_width = int(round((width * scaler) / divisibility) * divisibility)
new_height = int(round((height * scaler) / divisibility) * divisibility)
return {"width": new_width, "height": new_height}