make sure to set tokenizer from l3 70b and save safetensors

axolotl-ai-cloud · Jan 8, 2025 · 4df078a · 4df078a
1 parent 60e3a63
commit 4df078a
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tests/e2e/integrations/test_kd.py b/tests/e2e/integrations/test_kd.py
@@ -16,7 +16,8 @@
 @pytest.fixture(name="kd_min_cfg")
 def min_cfg(temp_dir):
     return {
-        "base_model": "unsloth/Llama-3.2-1B",
+        "base_model": "osllmai-community/Llama-3.2-1B",
+        "tokenizer_config": "osllmai-community/Llama-3.3-70B-Instruct",
         "plugins": [
             "axolotl.integrations.kd.KDPlugin",
             "axolotl.integrations.liger.LigerPlugin",
@@ -62,6 +63,7 @@ def min_cfg(temp_dir):
         },
         "max_steps": 5,
         "output_dir": temp_dir,
+        "save_safetensors": True,
     }