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

2026-04-25 16:59:22 +00:00 · 2023-09-24 05:19:57 -06:00
parent 830e87cb87
commit e5153d87c9
5 changed files with 41 additions and 20 deletions
--- a/toolkit/data_loader.py
+++ b/toolkit/data_loader.py
@@ -281,7 +281,8 @@ class PairedImageDataset(Dataset):
            bucket_resolution = get_bucket_for_image_size(
                width=img2.width,
                height=img2.height,
-                resolution=self.size
+                resolution=self.size,
+                # divisibility=self.
            )

            # images will be same base dimension, but may be trimmed. We need to shrink and then central crop