fix tests

lucidrains · Jan 14, 2024 · 298a0db · 298a0db
1 parent 9bcba1d
commit 298a0db
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.github/workflows/python-package.yml b/.github/workflows/python-package.yml
@@ -26,7 +26,7 @@ jobs:
     - name: Install dependencies
       run: |
         python -m pip install --upgrade pip
-        python -m pip install pytest numpy torch==1.10.0
+        python -m pip install pytest torch==2.0.1+cpu -f https://download.pytorch.org/whl/torch_stable.html
     - name: Test with pytest
       run: |
         python setup.py test