diff --git a/requirements.txt b/requirements.txt
index 061749902..2fa9ed935 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -26,7 +26,7 @@ sentencepiece
 gradio==3.50.2
 
 modal==0.70.5
-pydantic==2.6.3
+pydantic==2.10.6
 addict
 fire
 PyYAML>=6.0
diff --git a/setup.py b/setup.py
index 370eb7297..6a5c09aba 100644
--- a/setup.py
+++ b/setup.py
@@ -153,5 +153,8 @@ setup(
         "ray": [
             "ray[train]",
         ],
+        "vllm": [
+            "vllm>=0.7.1",
+        ],
     },
 )