Disable xformers when it can't load properly.

7339479b · comfyanonymous · f12ec559 · 7339479b
Commit 7339479b authored Nov 13, 2023 by comfyanonymous
Show whitespace changes
Inline Side-by-side

Showing with 4 additions and 0 deletions

comfy/model_management.py comfy/model_management.py +4 -0

No files found.
--- a/comfy/model_management.py
+++ b/comfy/model_management.py
@@ -133,6 +133,10 @@ else:
        import xformers
        import xformers.ops
        XFORMERS_IS_AVAILABLE = True
+        try:
+            XFORMERS_IS_AVAILABLE = xformers._has_cpp_library
+        except:
+            pass
        try:
            XFORMERS_VERSION = xformers.version.__version__
            print("xformers version:", XFORMERS_VERSION)