use smaller pretrained models for ci

2026-04-23 13:51:01 +00:00
parent 1bf65c500e
commit 431888c1de
24 changed files with 614 additions and 205 deletions
--- a/tests/e2e/test_qwen.py
+++ b/tests/e2e/test_qwen.py
@@ -17,7 +17,7 @@ class TestE2eQwen:
    Test cases for qwen models
    """

-    @pytest.mark.parametrize("base_model", ["Qwen/Qwen2-0.5B", "Qwen/Qwen2.5-0.5B"])
+    @pytest.mark.parametrize("base_model", ["axolotl-ai-co/tiny-qwen2-129m"])
    def test_dpo(self, base_model, temp_dir):
        cfg = DictDefault(
            {