more train steps

2026-04-25 02:17:48 +00:00
parent f77408a3d0
commit 1b9520cc8b
8 changed files with 105 additions and 49 deletions
--- a/tests/e2e/patched/test_phi_multipack.py
+++ b/tests/e2e/patched/test_phi_multipack.py
@@ -60,6 +60,7 @@ class TestPhiMultipack(unittest.TestCase):
                "bf16": "auto",
                "save_first_step": False,
                "use_tensorboard": True,
+                "seed": 42,
            }
        )

@@ -120,6 +121,7 @@ class TestPhiMultipack(unittest.TestCase):
                "bf16": "auto",
                "save_first_step": False,
                "use_tensorboard": True,
+                "seed": 42,
            }
        )