Enable pytorch attention by default on xpu.

321c5fa2 · comfyanonymous · 0665749b · 321c5fa2
Commit 321c5fa2 authored Sep 17, 2023 by comfyanonymous
Show whitespace changes
Inline Side-by-side

Showing with 3 additions and 0 deletions

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

No files found.
--- a/comfy/model_management.py
+++ b/comfy/model_management.py
@@ -165,6 +165,9 @@ try:
                ENABLE_PYTORCH_ATTENTION = True
            if torch.cuda.is_bf16_supported():
                VAE_DTYPE = torch.bfloat16
+    if is_intel_xpu():
+        if args.use_split_cross_attention == False and args.use_quad_cross_attention == False:
+            ENABLE_PYTORCH_ATTENTION = True
 except:
    pass