upgrade pytorch to 2.5.1 (#2024)

2024-11-08 10:46:24 -05:00
parent 035e9f9dd7
commit 3cb2d75de1
5 changed files with 9 additions and 9 deletions
--- a/.github/workflows/tests.yml
+++ b/.github/workflows/tests.yml
@@ -36,7 +36,7 @@ jobs:
      fail-fast: false
      matrix:
        python_version: ["3.10", "3.11"]
-        pytorch_version: ["2.3.1", "2.4.1", "2.5.0"]
+        pytorch_version: ["2.3.1", "2.4.1", "2.5.1"]
    timeout-minutes: 20

    steps:
@@ -132,7 +132,7 @@ jobs:
          - cuda: 124
            cuda_version: 12.4.1
            python_version: "3.11"
-            pytorch: 2.5.0
+            pytorch: 2.5.1
            num_gpus: 1
            axolotl_extras:
    steps: