tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

fix xformers version (#2888)

Browse Source
This commit is contained in:
Wing Lian
2025-07-09 08:43:40 -04:00
committed by GitHub
parent 89e99eaaa7
commit 4ff96a2526
2 changed files with 6 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
.github/workflows/tests.yml vendored
View File

@@ -52,7 +52,7 @@ jobs:
fail-fast: false fail-fast: false
matrix: matrix:
python_version: ["3.11"] python_version: ["3.11"]
pytorch_version: ["2.5.1", "2.6.0", "2.7.1"] pytorch_version: ["2.5.1", "2.6.0", "2.7.0", "2.7.1"]
timeout-minutes: 20 timeout-minutes: 20
steps: steps:
@@ -125,7 +125,7 @@ jobs:
fail-fast: false fail-fast: false
matrix: matrix:
python_version: ["3.11"] python_version: ["3.11"]
pytorch_version: ["2.5.1", "2.6.0", "2.7.1"] pytorch_version: ["2.5.1", "2.6.0", "2.7.0", "2.7.1"]
timeout-minutes: 20 timeout-minutes: 20
steps: steps:

5
setup.py
View File

@@ -66,7 +66,10 @@ def parse_requirements(extras_require_map):
if (major, minor) >= (2, 7): if (major, minor) >= (2, 7):
_install_requires.pop(_install_requires.index(xformers_version)) _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"] extras_require_map["vllm"] = ["vllm>=0.9.0"]
elif (major, minor) >= (2, 6): elif (major, minor) >= (2, 6):
_install_requires.pop(_install_requires.index(xformers_version)) _install_requires.pop(_install_requires.index(xformers_version))