removing 2.3.1 (#2294)

2025-01-29 04:23:44 +00:00
parent c015a76a23
commit c071a530f7
12 changed files with 10 additions and 114 deletions
--- a/.github/workflows/tests.yml
+++ b/.github/workflows/tests.yml
@@ -49,7 +49,7 @@ jobs:
      max-parallel: 2
      matrix:
        python_version: ["3.10", "3.11"]
-        pytorch_version: ["2.3.1", "2.4.1", "2.5.1"]
+        pytorch_version: ["2.4.1", "2.5.1"]
        exclude:
          - python_version: "3.10"
            pytorch_version: "2.4.1"
@@ -244,12 +244,6 @@ jobs:
      fail-fast: false
      matrix:
        include:
-          - cuda: 121
-            cuda_version: 12.1.1
-            python_version: "3.10"
-            pytorch: 2.3.1
-            num_gpus: 1
-            axolotl_extras: mamba-ssm
          - cuda: 124
            cuda_version: 12.4.1
            python_version: "3.11"