no evals for mamba tests

axolotl-ai-cloud · Dec 9, 2023 · 937d8e8 · 937d8e8
1 parent 046cd3b
commit 937d8e8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/e2e/test_mamba.py b/tests/e2e/test_mamba.py
@@ -36,7 +36,7 @@ def test_fft(self, temp_dir):
                 "flash_attention": False,
                 "sequence_len": 1024,
                 "load_in_8bit": False,
-                "val_set_size": 0.1,
+                "val_set_size": 0.0,
                 "datasets": [
                     {
                         "path": "mhenrichsen/alpaca_2k_test",
@@ -53,7 +53,7 @@ def test_fft(self, temp_dir):
                 "lr_scheduler": "cosine",
                 "max_steps": 20,
                 "save_steps": 10,
-                "eval_steps": 10,
+                "eval_steps": None,
                 "save_safetensors": False,
             }
         )