ComfyUI Node: TrainDatasetAdd

Authored by kijai

Created

Updated

784 stars

Category

FluxTrainer

Inputs

dataset_config JSON
width INT
height INT
batch_size INT
dataset_path STRING
class_tokens STRING
enable_bucket BOOLEAN
bucket_no_upscale BOOLEAN
num_repeats INT
min_bucket_reso INT
max_bucket_reso INT
regularization JSON

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