convert rgba to rgb (#1678)

pytorch · Sep 25, 2024 · e1c2fec · e1c2fec
1 parent 645e041
commit e1c2fec
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/torchtune/models/clip/_transform.py b/torchtune/models/clip/_transform.py
@@ -163,6 +163,8 @@ def __call__(
         assert isinstance(image, Image.Image), "Input image must be a PIL image."
 
         # Make image torch.tensor((3, H, W), dtype=dtype), 0<=values<=1
+        if hasattr(image, "mode") and image.mode == "RGBA":
+            image = image.convert("RGB")
         image = F.to_image(image)
         image = F.grayscale_to_rgb_image(image)
         image = F.to_dtype(image, dtype=self.dtype, scale=True)