Fix the vlm test (#1949)

The recent Flux model graudate PR breaks vlm

Author: fegin

tests/assets/cc12m_test/pack_test_dataset.py

@@ -53,9 +53,7 @@ def pack_wds_dataset(tar_destination, source_folder, number_of_samples):
 
 
 if __name__ == "__main__":
-    tar_destination = (
-        "torchtitan/experiments/flux/tests/assets/cc12m_test/cc12m-train-0000.tar"
-    )
+    tar_destination = "tests/assets/cc12m_test/cc12m-train-0000.tar"
     source_folder = "cc12m_test"
     number_of_samples = 32
     pack_wds_dataset(tar_destination, source_folder, number_of_samples)

torchtitan/experiments/vlm/datasets/mm_datasets.py

@@ -201,7 +201,7 @@ def _process_cc12_wd_sample(
     ),
     "cc12m-test": DatasetConfig(
         # TODO: move test cc12m dataset to core test folder
-        path="torchtitan/experiments/flux/tests/assets/cc12m_test",
+        path="tests/assets/cc12m_test",
         loader=lambda path: load_dataset(
             path, split="train", data_files={"train": "*.tar"}, streaming=True
         ),