fix xformers version (#2888)

2025-07-09 08:43:40 -04:00
parent 89e99eaaa7
commit 4ff96a2526
2 changed files with 6 additions and 3 deletions
--- a/setup.py
+++ b/setup.py
@@ -66,7 +66,10 @@ def parse_requirements(extras_require_map):

            if (major, minor) >= (2, 7):
                _install_requires.pop(_install_requires.index(xformers_version))
-                _install_requires.append("xformers==0.0.31.post1")
+                if patch == 0:
+                    _install_requires.append("xformers==0.0.30")
+                else:
+                    _install_requires.append("xformers==0.0.31.post1")
                extras_require_map["vllm"] = ["vllm>=0.9.0"]
            elif (major, minor) >= (2, 6):
                _install_requires.pop(_install_requires.index(xformers_version))