clean

nikg4 · nikg4 · commit ad3d69b1ae71 · 2024-09-27T14:10:00.000-07:00
diff --git a/src/oumi/datasets/vision_language/coco_captions.py b/src/oumi/datasets/vision_language/coco_captions.py
@@ -5,8 +5,7 @@
 
 @register_dataset("coco_captions")
 class COCOCaptionsDataset(VisionLanguageSftDataset):
-    # default_dataset = "HuggingFaceM4/COCO"
-    default_dataset = "/home/user/data/coco_captions/train"
+    default_dataset = "HuggingFaceM4/COCO"
     default_prompt = "Describe this image:"
 
     def transform_conversation(self, example: dict) -> Conversation:
