From 8d288a2ad48aefca02e2fc9773b1e63859a39411 Mon Sep 17 00:00:00 2001
From: Wing Lian <wing.lian@gmail.com>
Date: Fri, 13 Oct 2023 11:19:30 -0400
Subject: [PATCH] workaround for installing xformers w torch 2.1.0 (#725)

---
 setup.py | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/setup.py b/setup.py
index 3109d4af1a..1ceda5c044 100644
--- 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