fix merge conflict of duplicate max_steps in config for relora (#2116)

2024-12-03 07:42:41 -05:00
parent b9bb02406a
commit fc6188cd76
1 changed files with 0 additions and 1 deletions
--- a/tests/e2e/test_relora_llama.py
+++ b/tests/e2e/test_relora_llama.py
@@ -58,7 +58,6 @@ class TestReLoraLlama(unittest.TestCase):
                "output_dir": temp_dir,
                "learning_rate": 0.00001,
                "optimizer": "adamw_torch",
-                "max_steps": 5,
                "lr_scheduler": "cosine",
            }
        )