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_mistral_samplepack.py
+++ b/tests/e2e/patched/test_mistral_samplepack.py
@@ -63,6 +63,7 @@ class TestMistral(unittest.TestCase):
                "bf16": "auto",
                "save_first_step": False,
                "use_tensorboard": True,
+                "seed": 42,
            }
        )
        cfg = validate_config(cfg)
@@ -113,6 +114,7 @@ class TestMistral(unittest.TestCase):
                "bf16": "auto",
                "save_first_step": False,
                "use_tensorboard": True,
+                "seed": 42,
            }
        )
        cfg = validate_config(cfg)