Add RGB mode support for image loading in load_image function

Author: Ankur-singh

torchtune/data/_utils.py

@@ -78,15 +78,16 @@ def load_image(image_loc: Union[Path, str]) -> torch.Tensor:
         try:
             image_loc = request.urlopen(image_loc).read()
             image = torchvision.io.decode_image(
-                torch.frombuffer(image_loc, dtype=torch.uint8)
+                torch.frombuffer(image_loc, dtype=torch.uint8),
+                mode="RGB",
             )
         except Exception as e:
             raise ValueError("Failed to load remote image as torch.Tensor") from e
 
     # Open the local image as a Tensor image
     else:
         try:
-            image = torchvision.io.decode_image(image_loc)
+            image = torchvision.io.decode_image(image_loc, mode="RGB")
         except Exception as e:
             raise ValueError("Failed to load local image as torch.Tensor") from e
     return image