From f8a12ecc7f7ebcffe26e1ae405c6aa533fc400cd Mon Sep 17 00:00:00 2001
From: Roger Wang <136131678+ywang96@users.noreply.github.com>
Date: Sun, 24 Mar 2024 06:32:45 -0700
Subject: [PATCH] [Misc] Bump transformers version (#3592)

---
 requirements-rocm.txt | 4 ++--
 requirements.txt      | 2 +-
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/requirements-rocm.txt b/requirements-rocm.txt
index 07d94cd9..6acf7069 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 e136defa..eb9977d9 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]