diff --git a/tests/e2e/test_relora_llama.py b/tests/e2e/test_relora_llama.py index 70310923dc..5de5db11b7 100644 --- a/tests/e2e/test_relora_llama.py +++ b/tests/e2e/test_relora_llama.py @@ -58,7 +58,6 @@ def test_relora(self, temp_dir): "output_dir": temp_dir, "learning_rate": 0.00001, "optimizer": "adamw_torch", - "max_steps": 5, "lr_scheduler": "cosine", } )