make sure to download fixtures for kd test (#2541)

* make sure to download fixtures for kd test * use same alpaca dataset
2025-04-21 10:31:50 -04:00
parent 341e95aac9
commit 7651550850
6 changed files with 36 additions and 4 deletions
--- a/tests/e2e/solo/test_flex.py
+++ b/tests/e2e/solo/test_flex.py
@@ -41,8 +41,9 @@ class TestPackedFlex(unittest.TestCase):
                },
                "datasets": [
                    {
-                        "path": "vicgalle/alpaca-gpt4",
+                        "path": "tatsu-lab/alpaca",
                        "type": "alpaca",
+                        "split": "train[:10%]",
                    },
                ],
                "num_epochs": 1,