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

workaround for installing xformers w torch 2.1.0 (#725)

Browse Source
This commit is contained in:
Wing Lian
2023-10-13 11:19:30 -04:00
committed by GitHub
parent f30afe4544
commit 8d288a2ad4
1 changed files with 8 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
setup.py
View File

@@ -21,6 +21,14 @@ def parse_requirements():
): ):
# Handle standard packages # Handle standard packages
_install_requires.append(line) _install_requires.append(line)
# TODO(wing) remove once xformers release supports torch 2.1.0
if "torch==2.1.0" in _install_requires:
_install_requires.pop(_install_requires.index("xformers>=0.0.22"))
_install_requires.append(
"git+https://github.com/facebookresearch/xformers.git@main#egg=xformers"
)
return _install_requires, _dependency_links return _install_requires, _dependency_links