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_falcon_samplepack.py
+++ b/tests/e2e/patched/test_falcon_samplepack.py
@@ -61,6 +61,7 @@ class TestFalconPatched(unittest.TestCase):
                "bf16": "auto",
                "save_first_step": False,
                "use_tensorboard": True,
+                "seed": 42,
            }
        )
        cfg = validate_config(cfg)
@@ -110,6 +111,7 @@ class TestFalconPatched(unittest.TestCase):
                "bf16": "auto",
                "save_first_step": False,
                "use_tensorboard": True,
+                "seed": 42,
            }
        )
        cfg = validate_config(cfg)