fix ci

Author: lucidrains

.github/workflows/test.yml

@@ -17,7 +17,7 @@ jobs:
         run: |
           python -m pip install uv
           python -m uv pip install --upgrade pip
-          python -m uv pip install torch --index-url https://download.pytorch.org/whl/nightly/cpu
+          python -m uv pip install torch==2.8 --index-url https://download.pytorch.org/whl/nightly/cpu
           python -m uv pip install -e .[test]
       - name: Test with pytest
         run: |