ComfyUI Node: TrainDatasetAdd
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
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
Looking for a different node?
More nodes in ComfyUI Flux Trainer
Other FluxTrainer nodes
- Flux Train Validation Settings
- Init Flux LoRA Training
- Init Flux Training
- Optimizer Config
- Optimizer Config Adafactor
- Optimizer Config Prodigy
- Optimizer Config ProdigyPlusScheduleFree
- SD3 Train Validation Settings
- TrainDatasetGeneralConfig
- Train Dataset Regularization
- Train Network Config
- Upload To HuggingFace
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.