Update deps 202401 (#1204) [skip ci]

* update deps * xformers fix too
2024-01-25 10:11:49 -05:00
parent 33e117088f
commit a01b998c0f
2 changed files with 4 additions and 6 deletions
--- a/setup.py
+++ b/setup.py
@@ -27,9 +27,9 @@ def parse_requirements():

    try:
        torch_version = version("torch")
-        if torch_version.startswith("2.1.1"):
+        if torch_version.startswith("2.1."):
            _install_requires.pop(_install_requires.index("xformers==0.0.22"))
-            _install_requires.append("xformers==0.0.23")
+            _install_requires.append("xformers>=0.0.23")
    except PackageNotFoundError:
        pass