Swap Crop (#81)

Fix crop bug
mosaicml · Oct 12, 2023 · dc51257 · dc51257
1 parent 4a97a46
commit dc51257
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/diffusion/datasets/image_caption.py b/diffusion/datasets/image_caption.py
@@ -211,7 +211,7 @@ def build_streaming_image_caption_dataloader(
         sdxl = False
 
     # Setup the transforms to apply
-    crop_transform = LargestCenterSquare(resize_size) if rand_crop else RandomCropSquare(resize_size)
+    crop_transform = RandomCropSquare(resize_size) if rand_crop else LargestCenterSquare(resize_size)
     if transform is None:
         if sdxl:
             # Crop will return parameters so do separately