ComfyUI Node: Flux Trainer Loss Config

Authored by kijai

Created

Updated

786 stars

Category

FluxTrainer

Inputs

loss_type
  • l2
  • huber
  • smooth_l1
huber_schedule
  • snr
  • exponential
  • constant
huber_c FLOAT
huber_scale FLOAT

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