From 56f3b9f20fe4b93d45e7e645ae2be5a705ea416f Mon Sep 17 00:00:00 2001
From: Wing Lian <wing@axolotl.ai>
Date: Mon, 3 Feb 2025 20:02:18 -0500
Subject: [PATCH] bump pydantic to support vllm

---
 requirements.txt | 2 +-
 setup.py         | 3 +++
 2 files changed, 4 insertions(+), 1 deletion(-)

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",
+        ],
     },
 )