wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-07-21 23:41:28 +08:00
Code Issues Actions 10 Packages Projects Releases Wiki Activity

Accept workflows from the command line

Browse Source
This commit is contained in:
Benjamin Berman
2025-05-07 14:53:39 -07:00
parent da2cbf7c91
commit b6d3f1fb08
35 changed files with 459 additions and 319 deletions
Download Patch File Download Diff File Expand all files Collapse all files
comfy/ldm/modules/attention.py
+6 -6
View File
@@ -600,23 +600,23 @@ def attention_flash(q, k, v, heads, mask=None, attn_precision=None, skip_reshape
optimized_attention = attention_basic
if model_management.sage_attention_enabled():
logger.info("Using sage attention")
logger.debug("Using sage attention")
optimized_attention = attention_sage
elif model_management.xformers_enabled():
logger.info("Using xformers attention")
logger.debug("Using xformers attention")
optimized_attention = attention_xformers
elif model_management.flash_attention_enabled():
logging.info("Using Flash Attention")
logging.debug("Using Flash Attention")
optimized_attention = attention_flash
elif model_management.pytorch_attention_enabled():
logger.info("Using pytorch attention")
logger.debug("Using pytorch attention")
optimized_attention = attention_pytorch
else:
if args.use_split_cross_attention:
logger.info("Using split optimization for attention")
logger.debug("Using split optimization for attention")
optimized_attention = attention_split
else:
logger.info("Using sub quadratic optimization for attention, if you have memory or speed issues try using: --use-split-cross-attention")
logger.debug("Using sub quadratic optimization for attention, if you have memory or speed issues try using: --use-split-cross-attention")
optimized_attention = attention_sub_quad
optimized_attention_masked = optimized_attention