Unverified Commit e46fa5d5 authored by Allen.Dou's avatar Allen.Dou Committed by GitHub
Browse files

Restrict prometheus_client >= 0.18.0 to prevent errors when importing pkgs (#3070)

parent a8683102
......@@ -6,4 +6,4 @@ neuronx-cc
fastapi
uvicorn[standard]
pydantic >= 2.0 # Required for OpenAI server.
prometheus_client
prometheus_client >= 0.18.0
......@@ -10,4 +10,4 @@ transformers >= 4.38.0 # Required for Gemma.
fastapi
uvicorn[standard]
pydantic >= 2.0 # Required for OpenAI server.
prometheus_client
prometheus_client >= 0.18.0
......@@ -9,7 +9,7 @@ xformers == 0.0.23.post1 # Required for CUDA 12.1.
fastapi
uvicorn[standard]
pydantic >= 2.0 # Required for OpenAI server.
prometheus_client
prometheus_client >= 0.18.0
pynvml == 11.5.0
triton >= 2.1.0
cupy-cuda12x == 12.1.0 # Required for CUDA graphs. CUDA 11.8 users should install cupy-cuda11x instead.
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment