RomanNum3ral
/
ComfyUI
mirror of https://github.com/Comfy-Org/ComfyUI.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Merge 4a23bdc376 into 37ac9ff44f

This commit is contained in:
野生の男 2026-08-15 20:51:44 +02:00 committed by GitHub
parent 37ac9ff44f 4a23bdc376
commit b4e23d7513
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -51,7 +51,9 @@ except ImportError:
logging.error(f"\n\nTo use the `--use-flash-attention` feature, the `flash-attn` package must be installed first.\ncommand:\n\t{sys.executable} -m pip install flash-attn") logging.error(f"\n\nTo use the `--use-flash-attention` feature, the `flash-attn` package must be installed first.\ncommand:\n\t{sys.executable} -m pip install flash-attn")
exit(-1) exit(-1)
COMFY_KITCHEN_INT8_ATTENTION_IS_AVAILABLE = comfy_kitchen.int8_attention_is_available() COMFY_KITCHEN_INT8_ATTENTION_IS_AVAILABLE = (
torch.version.hip is None and comfy_kitchen.int8_attention_is_available()
)
REGISTERED_ATTENTION_FUNCTIONS = {} REGISTERED_ATTENTION_FUNCTIONS = {}
def register_attention_function(name: str, func: Callable): def register_attention_function(name: str, func: Callable):