From 1a2276402f16215cfd388ed9485d67b267e467d8 Mon Sep 17 00:00:00 2001 From: Lewis <1657236+ctjlewis@users.noreply.github.com> Date: Tue, 4 Feb 2025 13:12:22 -0600 Subject: [PATCH] =?UTF-8?q?=F0=9F=93=8C=20vLLM=20>=3D=200.7.1=20for=20devi?= =?UTF-8?q?ce=20fix=20(#2766)?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit see https://github.com/huggingface/trl/issues/2745 --- setup.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/setup.py b/setup.py index 22e8d6c2cb..d70406d67e 100644 --- a/setup.py +++ b/setup.py @@ -91,7 +91,7 @@ "quantization": ["bitsandbytes"], "scikit": ["scikit-learn"], "test": ["parameterized", "pytest-cov", "pytest-rerunfailures", "pytest-xdist", "pytest"], - "vllm": ["vllm; sys_platform != 'win32'"], # vllm is not available on Windows + "vllm": ["vllm>=0.7.1; sys_platform != 'win32'"], # vllm is not available on Windows "vlm": ["Pillow"], } EXTRAS["dev"] = []