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
|