Bump transformers and add vllm (#1405)

SilasMarvin · web-flow · commit f2f550641e93 · 2024-04-12T14:39:28.000-07:00
diff --git 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
@@ -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
