From 4e64549d4e17315caefbf382a924f3bb39d0392d Mon Sep 17 00:00:00 2001
From: Wing Lian <wing.lian@gmail.com>
Date: Wed, 30 Oct 2024 14:51:10 -0400
Subject: [PATCH] reduce number of steps

---
 tests/e2e/multigpu/test_llama.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/e2e/multigpu/test_llama.py b/tests/e2e/multigpu/test_llama.py
index 7ab84175d9..91cb80c8b3 100644
--- a/tests/e2e/multigpu/test_llama.py
+++ b/tests/e2e/multigpu/test_llama.py
@@ -631,7 +631,7 @@ def test_8bit_lora_ds_zero3(self, temp_dir):
                     },
                 ],
                 "num_epochs": 1,
-                "max_steps": 50,
+                "max_steps": 15,
                 "micro_batch_size": 4,
                 "gradient_accumulation_steps": 4,
                 "output_dir": temp_dir,