Fix Transorm -> Transform

mosaicml · Aug 20, 2024 · 054e853 · 054e853
1 parent f339ebb
commit 054e853
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/diffusion/datasets/image_caption_latents.py b/diffusion/datasets/image_caption_latents.py
@@ -14,7 +14,7 @@
 from torch.utils.data import DataLoader
 from torchvision import transforms
 
-from diffusion.datasets.laion.transforms import LargestCenterSquare, RandomCropAspectRatioTransorm, RandomCropSquare
+from diffusion.datasets.laion.transforms import LargestCenterSquare, RandomCropAspectRatioTransform, RandomCropSquare
 from diffusion.datasets.utils import make_streams
 
 log = logging.getLogger(__name__)
@@ -212,7 +212,7 @@ def build_streaming_image_caption_latents_dataloader(
     elif crop_type == 'random':
         crop = RandomCropSquare(resize_size)
     elif crop_type == 'aspect_ratio':
-        crop = RandomCropAspectRatioTransorm(resize_size, ar_bucket_boundaries)  # type: ignore
+        crop = RandomCropAspectRatioTransform(resize_size, ar_bucket_boundaries)  # type: ignore
     else:
         crop = None