From 7f2027d93f571ceffd56db79b66cf2c50f4c68e2 Mon Sep 17 00:00:00 2001
From: Wing Lian <wing.lian@gmail.com>
Date: Fri, 13 Oct 2023 15:21:17 -0400
Subject: [PATCH] tweak for xformers install w pytorch 2.1.0 (#727)

---
 setup.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index 1ceda5c04..ada5fcb28 100644
--- 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