wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-26 06:10:15 +08:00
Code Issues Actions 12 Packages Projects Releases Wiki Activity

Merge branch 'comfyanonymous:master' into master

Browse Source
This commit is contained in:
patientx 2025-05-24 14:00:54 +03:00 committed by GitHub
commit 9790aaac7b
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -20,8 +20,11 @@ if model_management.xformers_enabled():
if model_management.sage_attention_enabled(): if model_management.sage_attention_enabled():
try: try:
from sageattention import sageattn from sageattention import sageattn
except ModuleNotFoundError: except ModuleNotFoundError as e:
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") 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")
else:
raise e
exit(-1) exit(-1)
if model_management.flash_attention_enabled(): if model_management.flash_attention_enabled():