ComfyUI Node: NNT DefineReformer Attention
Run ComfyUI workflows without the setup
No installs, no CUDA version roulette, no GPU sitting idle on your bill. Bring a workflow and run it in the browser.
Category
NNT Neural Network Toolkit/Transformers
Inputs
embed_dim INT
num_heads INT
num_buckets INT
bucket_size INT
num_hashes INT
causal
- True
- False
dropout FLOAT
batch_first
- True
- False
LAYER_STACK LIST
Outputs
LIST
Extension: ComfyUI Neural Network Toolkit NNT
Neural Network Toolkit (NNT) for ComfyUI is an extensive set of custom ComfyUI nodes for designing, training, and fine-tuning neural networks. This toolkit allows defining models, layers, training workflows, transformers, and tensor operations in a visual manner using nodes.
Authored by inventorado
Looking for a different node?
More nodes in ComfyUI Neural Network Toolkit NNT
Other NNT Neural Network Toolkit/Transformers nodes
Run ComfyUI workflows without the setup
No installs, no CUDA version roulette, no GPU sitting idle on your bill. Bring a workflow and run it in the browser.