upgrade to latest deepspeed and make sure latest tagged axolotl images are using torch 2.8.0 (#3261)

2025-11-13 13:03:01 -05:00
parent dcf24fd24e
commit 301e22849f
2 changed files with 3 additions and 3 deletions
--- a/.github/workflows/main.yml
+++ b/.github/workflows/main.yml
@@ -25,7 +25,6 @@ jobs:
            python_version: "3.11"
            pytorch: 2.7.1
            axolotl_extras: vllm
-            is_latest: true
          - cuda: 128
            cuda_version: 12.8.1
            python_version: "3.11"
@@ -36,6 +35,7 @@ jobs:
            python_version: "3.11"
            pytorch: 2.8.0
            axolotl_extras:
+            is_latest: true
    runs-on: axolotl-gpu-runner
    steps:
      - name: Checkout
@@ -99,7 +99,6 @@ jobs:
            python_version: "3.11"
            pytorch: 2.7.1
            axolotl_extras: vllm
-            is_latest: true
          - cuda: 128
            cuda_version: 12.8.1
            python_version: "3.11"
@@ -110,6 +109,7 @@ jobs:
            python_version: "3.11"
            pytorch: 2.8.0
            axolotl_extras:
+            is_latest: true
    runs-on: axolotl-gpu-runner
    steps:
      - name: Checkout
--- a/setup.py
+++ b/setup.py
@@ -130,7 +130,7 @@ extras_require = {
        "ring-flash-attn>=0.1.7",
    ],
    "deepspeed": [
-        "deepspeed==0.17.5",
+        "deepspeed==0.18.2",
        "deepspeed-kernels",
    ],
    "mamba-ssm": [