fix tests

2026-04-23 23:47:28 +00:00
parent 5db4272f69
commit f77408a3d0
7 changed files with 42 additions and 20 deletions
--- a/tests/e2e/multigpu/test_fsdp2.py
+++ b/tests/e2e/multigpu/test_fsdp2.py
@@ -314,7 +314,8 @@ class TestFSDP2:
                "lora_alpha": 16,
                "lora_target_linear": True,
                "num_epochs": 1,
-                "max_steps": 20,
+                "max_steps": 30,
+                "warmup_steps": 3,
                "micro_batch_size": 2,
                "gradient_accumulation_steps": 1,
                "output_dir": temp_dir,