Patch Flash Attention DN

PatchFlashAttentionDN

DN/FlashAttentionexperimental

Patches ComfyUI attention to use Flash Attention 2 (flash_attn). Requires the flash_attn package. Wire the output MODEL into your sampler.

Pack: ComfyUI-DN_PatchFlashAttention

custom_nodes.ComfyUI-DN_PatchFlashAttention

Inputs (2)

NameTypeRequired
modelMODELrequired
enabledBOOLEANrequired

Outputs (1)

NameType
MODELMODEL
Patch Flash Attention DN | TealPug Node Explorer