ComfyUI Node: Train SAE

Authored by Nynxz

Created

Updated

4 stars

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

Concept Steer/Features

Inputs

save_path STRING
data_source COMBO
layer INT
sae_expansion INT
n_vectors INT
epochs INT
n_prompts INT
l1_coeff FLOAT
learning_rate FLOAT
cache_activations BOOLEAN
hf_dataset STRING
hf_subset STRING
encoder_path STRING
seed INT
protect_existing BOOLEAN

Outputs

STRING

Extension: Concept Steer

Concept Steering for ComfyUI — steer image generation with learned concept directions via SAE + DPO

Authored by Nynxz

Looking for a different node?

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.

Learn more