need to update deepspeed version in extras too

axolotl-ai-cloud · Dec 9, 2024 · fcddcc9 · fcddcc9
1 parent c78de6f
commit fcddcc9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -125,7 +125,7 @@ def get_package_version():
             "flash-attn==2.7.0.post2",
         ],
         "deepspeed": [
-            "deepspeed==0.15.4",
+            "deepspeed==0.16.1",
             "deepspeed-kernels",
         ],
         "mamba-ssm": [