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"] = []