diff --git a/setup.py b/setup.py
index 2a5b86500b..694493ce62 100644
--- a/setup.py
+++ b/setup.py
@@ -154,7 +154,7 @@ def package_files(prefix: str, directory: str, extension: str):
 }
 
 extra_deps['deepspeed'] = [
-    'numpy<2',
+    'numpy<3',
     'deepspeed==0.8.3',
     'pydantic>=1.0,<2',
 ]