A ComfyUI custom node that patches the attention mechanism to use Flash Attention 2, similar to how Patch Sage Attention KJ works in KJNodes.
https://github.com/0xDELUXA/ComfyUI-DN_PatchFlashAttention
git clone https://github.com/0xDELUXA/ComfyUI-DN_PatchFlashAttention
comfy node install dn_patchflashattention