ComfyUI Node: Ptn Embedding Transformer Linear
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
Training
Inputs
num_layers INT
vocabulary_size INT
hidden_size INT
nhead INT
dim_feedforward INT
dropout FLOAT
nonlinearity
- gelu
- relu
layer_norm_eps FLOAT
batch_first BOOLEAN
norm_first BOOLEAN
bias BOOLEAN
max_length INT
linear_output_size INT
linear_bias BOOLEAN
Outputs
PTMODEL
Extension: ComfyUI-Pt-Wrapper
PyTorch extension for ComfyUI featuring extensive PyTorch wrapper nodes for seamless tensor operations and PyTorch model training.
Authored by HowToSD
Looking for a different node?
More nodes in ComfyUI-Pt-Wrapper
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.