Skip to content

Commit 09d46ad

Browse files
committed
support vllm 0.14.0 for torch 2.9.1
1 parent 9c8d33c commit 09d46ad

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

setup.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -79,6 +79,10 @@ def parse_requirements(extras_require_map):
7979
if not install_xformers:
8080
_install_requires.pop(_install_requires.index(xformers_version))
8181
extras_require_map["vllm"] = ["vllm==0.13.0"]
82+
if patch == 0:
83+
extras_require_map["vllm"] = ["vllm==0.13.0"]
84+
else:
85+
extras_require_map["vllm"] = ["vllm==0.14.0"]
8286
elif (major, minor) >= (2, 8):
8387
extras_require_map.pop("fbgemm-gpu")
8488
extras_require_map["fbgemm-gpu"] = ["fbgemm-gpu-genai==1.3.0"]

0 commit comments

Comments
 (0)