xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2026-09-02 08:27:05 +08:00
Code Issues Packages Projects Releases Wiki Activity

Ruff fixes

Browse Source
This commit is contained in:
Panchovix 2025-07-25 21:18:42 -04:00 committed by GitHub
parent 2d69b2f064
commit bb7393565b
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
comfy/ldm/modules/attention.py
View File

@ -20,13 +20,13 @@ if model_management.xformers_enabled():
if model_management.sage_attention_enabled(): if model_management.sage_attention_enabled():
sage_attention_available = False sage_attention_available = False
SAGE_ATTENTION_3_AVAILABLE = False SAGE_ATTENTION_3_AVAILABLE = False
try: try:
from sageattn import sageattn_blackwell from sageattn import sageattn_blackwell
SAGE_ATTENTION_3_AVAILABLE = True SAGE_ATTENTION_3_AVAILABLE = True
sage_attention_available = True sage_attention_available = True
print("Found SageAttention3 (sageattn package)") print("Found SageAttention3 (sageattn package)")
except ImportError: except ImportError:
try: try:
from sageattention import sageattn from sageattention import sageattn
@ -34,12 +34,9 @@ if model_management.sage_attention_enabled():
print("Found SageAttention2 (sageattention package)") print("Found SageAttention2 (sageattention package)")
except ModuleNotFoundError as e: except ModuleNotFoundError as e:
pass pass
if not sage_attention_available: if not sage_attention_available:
if e.name == "sageattention": logging.error(f"\n\nTo use the `--use-sage-attention` feature, the `sageattention` package must be installed first.\ncommand:\n\t{sys.executable} -m pip install sageattention")
logging.error(f"\n\nTo use the `--use-sage-attention` feature, the `sageattention` package must be installed first.\ncommand:\n\t{sys.executable} -m pip install sageattention")
else:
raise e
exit(-1) exit(-1)
if model_management.flash_attention_enabled(): if model_management.flash_attention_enabled():