From 80dc4c261afb6a6ae5b3d383be4b65d9dbf517c4 Mon Sep 17 00:00:00 2001 From: Wing Lian Date: Mon, 14 Jul 2025 09:24:29 -0400 Subject: [PATCH] fix xformers version for python 2.6 (#2916) [skip ci] --- setup.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/setup.py b/setup.py index 731fe8a6f..ff8bd2c5c 100644 --- a/setup.py +++ b/setup.py @@ -73,9 +73,9 @@ def parse_requirements(extras_require_map): extras_require_map["vllm"] = ["vllm>=0.9.0"] elif (major, minor) >= (2, 6): _install_requires.pop(_install_requires.index(xformers_version)) - _install_requires.append( - "xformers==0.0.29.post2" - ) # vllm needs post2 w torch 2.6 + _install_requires.append("xformers==0.0.29.post3") + # since we only support 2.6.0+cu126 + _dependency_links.append("https://download.pytorch.org/whl/cu126") extras_require_map["vllm"] = ["vllm==0.8.5.post1"] elif (major, minor) >= (2, 5): _install_requires.pop(_install_requires.index(xformers_version))