From fc6188cd760cb2ef7221a9bdce83ff0330e0208d Mon Sep 17 00:00:00 2001 From: Wing Lian Date: Tue, 3 Dec 2024 07:42:41 -0500 Subject: [PATCH] fix merge conflict of duplicate max_steps in config for relora (#2116) --- tests/e2e/test_relora_llama.py | 1 - 1 file changed, 1 deletion(-) 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", } )