Fix bug getting h,w

mosaicml · Jul 20, 2024 · 6fabbf5 · 6fabbf5
1 parent 1150340
commit 6fabbf5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/diffusion/datasets/laion/transforms.py b/diffusion/datasets/laion/transforms.py
@@ -134,7 +134,7 @@ def __init__(
         self.log_aspect_ratio_buckets = torch.log(self.aspect_ratio_buckets)
 
     def __call__(self, img, aspect_ratio):
-        orig_h, orig_w = img.size
+        orig_w, orig_h = img.size
         orig_aspect_ratio = orig_h / orig_w
         # Figure out target H/W given the input aspect ratio
         bucket_ind = torch.abs(self.log_aspect_ratio_buckets - math.log(aspect_ratio)).argmin()