zlei6/ComfyUI
1
0
Fork 0
mirror of https://github.com/Comfy-Org/ComfyUI.git synced 2026-09-09 02:26:31 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix --use-flash-attention ignored when xformers installed. (#14083)

Browse Source
This commit is contained in:
comfyanonymous
2026-05-23 17:44:28 -07:00
committed by GitHub
parent 0af123022d
commit 08d809d128
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
comfy/ldm/modules/attention.py
+3 -3
View File
@@ -741,12 +741,12 @@ optimized_attention = attention_basic
if model_management.sage_attention_enabled(): if model_management.sage_attention_enabled():
logging.info("Using sage attention") logging.info("Using sage attention")
optimized_attention = attention_sage optimized_attention = attention_sage
elif model_management.xformers_enabled():
logging.info("Using xformers attention")
optimized_attention = attention_xformers
elif model_management.flash_attention_enabled(): elif model_management.flash_attention_enabled():
logging.info("Using Flash Attention") logging.info("Using Flash Attention")
optimized_attention = attention_flash optimized_attention = attention_flash
elif model_management.xformers_enabled():
logging.info("Using xformers attention")
optimized_attention = attention_xformers
elif model_management.pytorch_attention_enabled(): elif model_management.pytorch_attention_enabled():
logging.info("Using pytorch attention") logging.info("Using pytorch attention")
optimized_attention = attention_pytorch optimized_attention = attention_pytorch