diff --git a/requirements-rocm.txt b/requirements-rocm.txt index 07d94cd94f5fa..6acf70695cef8 100644 --- a/requirements-rocm.txt +++ b/requirements-rocm.txt @@ -7,9 +7,9 @@ ray >= 2.9 sentencepiece # Required for LLaMA tokenizer. numpy tokenizers>=0.15.0 -transformers >= 4.39.0 # Required for StarCoder2. +transformers >= 4.39.1 # Required for StarCoder2 & Llava. fastapi uvicorn[standard] pydantic >= 2.0 # Required for OpenAI server. prometheus_client >= 0.18.0 -outlines == 0.0.34 \ No newline at end of file +outlines == 0.0.34 diff --git a/requirements.txt b/requirements.txt index e136defad4943..eb9977d93dd8d 100644 --- a/requirements.txt +++ b/requirements.txt @@ -5,7 +5,7 @@ ray >= 2.9 sentencepiece # Required for LLaMA tokenizer. numpy torch == 2.1.2 -transformers >= 4.39.0 # Required for StarCoder2. +transformers >= 4.39.1 # Required for StarCoder2 & Llava. xformers == 0.0.23.post1 # Required for CUDA 12.1. fastapi uvicorn[standard]