ComfyUI Node: TrainDatasetGeneralConfig

Authored by kijai

Created

Updated

781 stars

Category

FluxTrainer

Inputs

color_aug BOOLEAN
flip_aug BOOLEAN
shuffle_caption BOOLEAN
caption_dropout_rate FLOAT
alpha_mask BOOLEAN
reset_on_queue BOOLEAN
caption_extension STRING

Outputs

JSON

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