ComfyUI Node: Optimizer Config
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
FluxTrainer
Inputs
optimizer_type
- adamw8bit
- adamw
- prodigy
- CAME
- Lion8bit
- Lion
- adamwschedulefree
- sgdschedulefree
- AdEMAMix8bit
- PagedAdEMAMix8bit
- ProdigyPlusScheduleFree
max_grad_norm FLOAT
lr_scheduler
- constant
- cosine
- cosine_with_restarts
- polynomial
- constant_with_warmup
lr_warmup_steps INT
lr_scheduler_num_cycles INT
lr_scheduler_power FLOAT
min_snr_gamma FLOAT
extra_optimizer_args STRING
Outputs
ARGS
Extension: ComfyUI Flux Trainer
Currently supports LoRA training, and untested full finetune with code from kohya's scripts: a/https://github.com/kohya-ss/sd-scripts
Authored by kijai
Looking for a different node?
More nodes in ComfyUI Flux Trainer
Other FluxTrainer 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.