diff --git a/setup.py b/setup.py
index 243853959b..70a5e92ac0 100644
--- a/setup.py
+++ b/setup.py
@@ -155,7 +155,7 @@ def package_files(prefix: str, directory: str, extension: str):
 
 extra_deps['deepspeed'] = [
     'numpy<2',
-    'deepspeed==0.8.3',
+    'deepspeed==0.15.4',
     'pydantic>=1.0,<2',
 ]