Add xformers requirement

06bea896 · Casper Hansen · f3695d60 · 06bea896
Commit 06bea896 authored Sep 09, 2023 by Casper Hansen
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

setup.py setup.py +2 -1

No files found.
--- a/setup.py
+++ b/setup.py
@@ -45,7 +45,8 @@ requirements = [
    "attributedict",
    "protobuf",
    "torchvision",
-    "tabulate"
+    "tabulate",
+    "xformers"
 ]

 def get_include_dirs():