slow fsdp1 test

triton versions for older pytorch
upgrade liger to 0.6.5 and triton to 3.5.1
2026-02-10 13:23:52 -05:00 · 2026-02-10 11:09:03 -05:00 · 2026-02-10 11:05:00 -05:00
2 changed files with 3 additions and 2 deletions
--- a/requirements.txt
+++ b/requirements.txt
@@ -2,10 +2,10 @@
 # START section of dependencies that don't install on Darwin/MacOS
 bitsandbytes==0.49.1
-triton>=3.0.0
+triton>=3.4.0
 mamba-ssm==1.2.0.post1
 xformers>=0.0.23.post1
-liger-kernel==0.6.4
+liger-kernel==0.6.5
 # END section
 packaging==26.0
--- a/tests/e2e/multigpu/test_fsdp1.py
+++ b/tests/e2e/multigpu/test_fsdp1.py
@@ -186,6 +186,7 @@ class TestFSDP1:
        verify_training_success(temp_dir)
    @pytest.mark.skip(reason="slow test, deprecate fsdp1 asap")
    def test_dpo_fft(self, temp_dir):
        cfg = DictDefault(
            {
Author	SHA1	Message	Date
Wing Lian	2d13a06722	slow fsdp1 test	2026-02-10 13:23:52 -05:00
Wing Lian	ba27e830e8	triton versions for older pytorch	2026-02-10 11:09:03 -05:00
Wing Lian	8f7219e139	upgrade liger to 0.6.5 and triton to 3.5.1	2026-02-10 11:05:00 -05:00