Merge branch 'master' into worksplit-multigpu

2025-03-15 09:12:09 -05:00
parent cc928a786d 55a1b09ddc
commit c4ba399475
7 changed files with 102 additions and 12 deletions
--- a/comfy/model_management.py
+++ b/comfy/model_management.py
@@ -958,6 +958,9 @@ def cast_to_device(tensor, device, dtype, copy=False):
 def sage_attention_enabled():
    return args.use_sage_attention

+def flash_attention_enabled():
+    return args.use_flash_attention
+
 def xformers_enabled():
    global directml_enabled
    global cpu_state