ComfyUI Node
Flex Attention (torch.compile)
Patch attention to use PyTorch Flex Attention (torch.compile). No extra dependencies — works on any GPU supported by torch.compile. Falls back to PyTorch SDPA when masks are present.
Flex Attention (torch.compile)
- model
- MODEL
Categoryattention
Inputs (1)
| Name | Type | Default | Description |
|---|---|---|---|
| model | MODEL | — |
Outputs (1)
| Name | Type | Description |
|---|---|---|
| MODEL | MODEL | — |