rebased hymba multipack

2024-12-11 14:52:05 -05:00
parent 42bd32a233
commit cce7007bf8
8 changed files with 278 additions and 2 deletions
--- a/tests/e2e/test_optimizers.py
+++ b/tests/e2e/test_optimizers.py
@@ -67,8 +67,8 @@ class TestCustomOptimizers(unittest.TestCase):
        train(cfg=cfg, cli_args=cli_args, dataset_meta=dataset_meta)
        assert (Path(temp_dir) / "adapter_model.bin").exists()

-    @with_temp_dir
    @require_torch_2_5_1
+    @with_temp_dir
    def test_adopt_adamw(self, temp_dir):
        # pylint: disable=duplicate-code
        cfg = DictDefault(