tweak for xformers install w pytorch 2.1.0 (#727)

axolotl-ai-cloud · Oct 13, 2023 · 7f2027d · 7f2027d
1 parent 8d288a2
commit 7f2027d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -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