ComfyUI Node: Optimizer Config ProdigyPlusScheduleFree

Authored by kijai

Created

Updated

784 stars

Category

FluxTrainer

Inputs

lr FLOAT
max_grad_norm FLOAT
prodigy_steps INT
d0 FLOAT
d_coeff FLOAT
split_groups BOOLEAN
use_bias_correction BOOLEAN
min_snr_gamma FLOAT
use_stableadamw BOOLEAN
use_cautious BOOLEAN
use_adopt BOOLEAN
use_grams BOOLEAN
stochastic_rounding BOOLEAN
use_orthograd BOOLEAN
use_focus BOOLEAN
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

Run ComfyUI workflows in the Cloud!

No downloads or installs are required. Pay only for active GPU usage, not idle time. No complex setups and dependency issues

Learn more