From e46fa5d52e02ee48d5fdd12b35e39993008b4bd6 Mon Sep 17 00:00:00 2001 From: "Allen.Dou" Date: Wed, 28 Feb 2024 13:38:26 +0800 Subject: [PATCH] Restrict prometheus_client >= 0.18.0 to prevent errors when importing pkgs (#3070) --- requirements-neuron.txt | 2 +- requirements-rocm.txt | 2 +- requirements.txt | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/requirements-neuron.txt b/requirements-neuron.txt index 36e629ad..858472c2 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 e759ba7d..53bd11de 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 de93ba63..d4599ec9 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.