add qwen 0.5b to pytest session fixture

axolotl-ai-cloud · Nov 29, 2024 · adce6fa · adce6fa
1 parent c4eaa52
commit adce6fa
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/tests/conftest.py b/tests/conftest.py
@@ -14,6 +14,12 @@ def download_smollm2_135m_model():
     snapshot_download("HuggingFaceTB/SmolLM2-135M")
 
 
+@pytest.fixture(scope="session", autouse=True)
+def download_qwen_2_5_half_billion_model():
+    # download the model
+    snapshot_download("Qwen/Qwen2.5-0.5B")
+
+
 @pytest.fixture(scope="session", autouse=True)
 def download_tatsu_lab_alpaca_dataset():
     # download the model