deprecate torch 2.7.1 (#3339)

2026-01-01 06:52:45 -05:00
parent 2b199f9915
commit afe18ace35
11 changed files with 65 additions and 144 deletions
--- a/.github/workflows/multi-gpu-e2e.yml
+++ b/.github/workflows/multi-gpu-e2e.yml
@@ -29,13 +29,6 @@ jobs:
      fail-fast: false
      matrix:
        include:
-          - cuda: 126
-            cuda_version: 12.6.3
-            python_version: "3.11"
-            pytorch: 2.7.1
-            axolotl_extras: vllm
-            num_gpus: 2
-            nightly_build: "true"
          - cuda: 128
            cuda_version: 12.8.1
            python_version: "3.11"
@@ -46,7 +39,7 @@ jobs:
          - cuda: 128
            cuda_version: 12.8.1
            python_version: "3.11"
-            pytorch: 2.9.0
+            pytorch: 2.9.1
            axolotl_extras: fbgemm-gpu
            num_gpus: 2
            nightly_build: "true"