--disable-xformers should not even try to import xformers.

2023-03-13 11:36:48 -04:00
parent ebfa749b7b
commit a256a2abde
4 changed files with 15 additions and 16 deletions
--- a/comfy/ldm/modules/attention.py
+++ b/comfy/ldm/modules/attention.py
@@ -11,11 +11,10 @@ from .sub_quadratic_attention import efficient_dot_product_attention

 import model_management

-try:
+
+if model_management.xformers_enabled():
    import xformers
    import xformers.ops
-except:
-    pass

 # CrossAttn precision handling
 import os