chore(setup.py): bump vllm>=0.7.1 (#34)

See huggingface/trl#2766.
SmallDoges · Feb 6, 2025 · 80e57b3 · 80e57b3
1 parent 8c2168f
commit 80e57b3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -63,7 +63,7 @@
     "torch>=2.5.1",
     "transformers @ git+https://github.com/LoserCheems/transformers.git@support-constant-lr-with-cooldown",
     "trl @ git+https://github.com/huggingface/trl.git@main",
-    "vllm>=0.7.0",
+    "vllm>=0.7.1",
     "wandb>=0.19.1",
 ]