ComfyUI Node
Patch Flash Attention DN
Patches ComfyUI attention to use Flash Attention 2 (flash_attn). Requires the flash_attn package. Wire the output MODEL into your sampler.
Patch Flash Attention DN
- model
- MODEL
◄enabledtrue►
CategoryDN/FlashAttention
Inputs (2)
| Name | Type | Default | Description |
|---|---|---|---|
| model | MODEL | — | |
| enabled | BOOLEAN | true | Enable Flash Attention 2 patch. Set to False to pass the model through unchanged. |
Outputs (1)
| Name | Type | Description |
|---|---|---|
| MODEL | MODEL | — |