diff --git a/presets/inference/text-generation/requirements.txt b/presets/inference/text-generation/requirements.txt index 93263e253..2530fed6a 100644 --- a/presets/inference/text-generation/requirements.txt +++ b/presets/inference/text-generation/requirements.txt @@ -5,7 +5,7 @@ transformers==4.41.2 torch==2.2.0 accelerate==0.30.1 fastapi>=0.111.0,<0.112.0 # Allow patch updates -pydantic==2.7.4 # Allow patch updates +pydantic==2.7.4 uvicorn[standard]>=0.29.0,<0.30.0 # Allow patch updates peft==0.11.1 numpy==1.22.4 diff --git a/presets/tuning/text-generation/requirements.txt b/presets/tuning/text-generation/requirements.txt index ae3e55925..e2aeb3098 100644 --- a/presets/tuning/text-generation/requirements.txt +++ b/presets/tuning/text-generation/requirements.txt @@ -3,7 +3,7 @@ transformers==4.41.2 torch==2.2.0 accelerate==0.30.1 fastapi>=0.111.0,<0.112.0 # Allow patch updates -pydantic==2.7.4 # Allow patch updates +pydantic==2.7.4 uvicorn[standard]>=0.29.0,<0.30.0 # Allow patch updates # Utility libraries