workaround for installing xformers w torch 2.1.0 (#725)

axolotl-ai-cloud · Oct 13, 2023 · 8d288a2 · 8d288a2
1 parent f30afe4
commit 8d288a2
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/setup.py b/setup.py
@@ -21,6 +21,14 @@ def parse_requirements():
             ):
                 # Handle standard packages
                 _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