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/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": [