Flash Attention 4 (flash_attn.cute) node for ComfyUI. Install by: git clone https://github.com/sorryhyun/flash-attention-sm120-fix.git && cd flash-attention-sm120-fix && pip install -e "flash_attn/cute[dev]"
https://github.com/sorryhyun/flash-attention-sm120-fix/tree/dz/sm120_tma_optimized
git clone https://github.com/sorryhyun/flash-attention-sm120-fix/tree/dz/sm120_tma_optimized
comfy node install comfyui-flash-attention4