From c09658e6a9ff9ec9a24c7d839f413cc449df28dc Mon Sep 17 00:00:00 2001
From: SilasMarvin <19626586+SilasMarvin@users.noreply.github.com>
Date: Fri, 12 Apr 2024 14:37:08 -0700
Subject: [PATCH] Bump transformers and add vllm

---
 pgml-extension/requirements.linux.txt | 3 ++-
 pgml-extension/requirements.macos.txt | 2 +-
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/pgml-extension/requirements.linux.txt b/pgml-extension/requirements.linux.txt
index c481b0636..166689ca9 100644
--- a/pgml-extension/requirements.linux.txt
+++ b/pgml-extension/requirements.linux.txt
@@ -129,7 +129,7 @@ torch==2.1.2
 torchaudio==2.1.2
 torchvision==0.16.2
 tqdm==4.66.1
-transformers==4.38.2
+transformers==4.39.3
 transformers-stream-generator==0.0.4
 triton==2.1.0
 trl==0.7.10
@@ -142,3 +142,4 @@ xformers==0.0.23.post1
 xgboost==2.0.2
 xxhash==3.4.1
 yarl==1.9.4
+vllm==0.4.0.post1
diff --git a/pgml-extension/requirements.macos.txt b/pgml-extension/requirements.macos.txt
index 3d6a1aa8f..4cd2dc81b 100644
--- a/pgml-extension/requirements.macos.txt
+++ b/pgml-extension/requirements.macos.txt
@@ -93,7 +93,7 @@ torch==2.1.2
 torchaudio==2.1.2
 torchvision==0.16.2
 tqdm==4.66.1
-transformers==4.38.2
+transformers==4.39.3
 transformers-stream-generator==0.0.4
 typing-inspect==0.9.0
 typing_extensions==4.9.0