Disable xformers when it can't load properly.

2023-11-13 12:27:44 -05:00 · 2023-11-13 12:27:44 -05:00 · 7339479b10
parent f12ec55983
commit 7339479b10
1 changed files with 4 additions and 0 deletions
--- 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)