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

tweak for xformers install w pytorch 2.1.0 (#727)

Browse Source
This commit is contained in:
Wing Lian
2023-10-13 15:21:17 -04:00
committed by GitHub
parent 8d288a2ad4
commit 7f2027d93f
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
setup.py
View File

@@ -26,7 +26,7 @@ def parse_requirements():
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"
"xformers @ git+https://github.com/facebookresearch/xformers.git@main"
)
return _install_requires, _dependency_links