diff --git a/requirements-neuron.txt b/requirements-neuron.txt index 36e629add664..858472c20ca8 100644 --- a/requirements-neuron.txt +++ b/requirements-neuron.txt @@ -6,4 +6,4 @@ neuronx-cc fastapi uvicorn[standard] pydantic >= 2.0 # Required for OpenAI server. -prometheus_client +prometheus_client >= 0.18.0 diff --git a/requirements-rocm.txt b/requirements-rocm.txt index e759ba7d028d..53bd11de7c9d 100644 --- a/requirements-rocm.txt +++ b/requirements-rocm.txt @@ -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 diff --git a/requirements.txt b/requirements.txt index de93ba6354cd..d4599ec95d94 100644 --- a/requirements.txt +++ b/requirements.txt @@ -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.