diff --git a/setup.py b/setup.py
index 26c44738e4..49f97afc07 100644
--- a/setup.py
+++ b/setup.py
@@ -156,7 +156,7 @@ def package_files(prefix: str, directory: str, extension: str):
 extra_deps['deepspeed'] = [
     'numpy<2',
     'deepspeed==0.8.3',
-    'pydantic>=1.0,<2',
+    'pydantic>=1.0,<3',
 ]
 
 extra_deps['wandb'] = [