tweak for xformers install w pytorch 2.1.0 (#727)
Browse files
setup.py
CHANGED
|
@@ -26,7 +26,7 @@ def parse_requirements():
|
|
| 26 |
if "torch==2.1.0" in _install_requires:
|
| 27 |
_install_requires.pop(_install_requires.index("xformers>=0.0.22"))
|
| 28 |
_install_requires.append(
|
| 29 |
-
"git+https://github.com/facebookresearch/xformers.git@main
|
| 30 |
)
|
| 31 |
|
| 32 |
return _install_requires, _dependency_links
|
|
|
|
| 26 |
if "torch==2.1.0" in _install_requires:
|
| 27 |
_install_requires.pop(_install_requires.index("xformers>=0.0.22"))
|
| 28 |
_install_requires.append(
|
| 29 |
+
"xformers @ git+https://github.com/facebookresearch/xformers.git@main"
|
| 30 |
)
|
| 31 |
|
| 32 |
return _install_requires, _dependency_links
|