ostris/ai-toolkit
1
0
Fork 0
mirror of https://github.com/ostris/ai-toolkit.git synced 2026-04-30 11:11:37 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fixed issues with dataloader bucketing. Allow using standard base image for t2i adapters.

Browse Source
This commit is contained in:
Jaret Burkett
2023-09-24 05:19:57 -06:00
parent 830e87cb87
commit e5153d87c9
5 changed files with 41 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
toolkit/dataloader_mixins.py
View File

@@ -121,7 +121,7 @@ class BucketsMixin:
width = file_item.crop_width
height = file_item.crop_height
bucket_resolution = get_bucket_for_image_size(width, height, resolution=resolution)
bucket_resolution = get_bucket_for_image_size(width, height, resolution=resolution, divisibility=bucket_tolerance)
# set the scaling height and with to match smallest size, and keep aspect ratio
if width > height: