ComfyUI Node
Ptn Multihead Attention Custom
A ComfyUI node in Training with 7 inputs and 1 output.
Ptn Multihead Attention Custom
- PTMODEL
◄embed_dim512►
◄num_heads8►
◄dropout0.00►
◄biastrue►
◄kdim512►
◄vdim512►
◄batch_firsttrue►
CategoryTraining
Inputs (7)
| Name | Type | Default | Description |
|---|---|---|---|
| embed_dim | INT | 5121–1000000 | — |
| num_heads | INT | 81–256 | — |
| dropout | FLOAT | 0.000–1 | — |
| bias | BOOLEAN | true | — |
| kdim | INT | 5121–1000000 | — |
| vdim | INT | 5121–1000000 | — |
| batch_first | BOOLEAN | true | — |
Outputs (1)
| Name | Type | Description |
|---|---|---|
| PTMODEL | PTMODEL | — |