bumping pytorch version again

HanGuo97 · Nov 17, 2024 · 679b1f2 · 679b1f2
1 parent a622d7b
commit 679b1f2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.github/workflows/wheels.yaml b/.github/workflows/wheels.yaml
@@ -17,7 +17,7 @@ jobs:
       matrix:
           os: ['ubuntu-20.04']
           python-version: ['3.8', '3.9', '3.10', '3.11']
-          pytorch-version: ['2.4.1']
+          pytorch-version: ['2.5.1']
           cuda-version: ['11.8', '12.1']
 
     steps: