Realtime LoRA Trainer

RealtimeLoraTrainer

Trains a LoRA on-the-fly from one or more images using AI-Toolkit. Connect output to ApplyTrainedLora node to apply. Supports Z-Image, FLUX, Wan, and Qwen models.

Pack: Realtime LoRA Trainer

custom_nodes.comfyUI-Realtime-Lora

Inputs (20)

NameTypeRequired
inputcountINTrequired
images_pathSTRINGrequired
architectureCOMBOrequired
ai_toolkit_pathSTRINGrequired
captionSTRINGrequired
training_stepsINTrequired
learning_rateFLOATrequired
lora_rankINTrequired
vram_modeCOMBOrequired
keep_loraBOOLEANrequired
output_nameSTRINGrequired
custom_python_exeSTRINGrequired
image_1IMAGEoptional
caption_1STRINGoptional
image_2IMAGEoptional
caption_2STRINGoptional
image_3IMAGEoptional
caption_3STRINGoptional
image_4IMAGEoptional
caption_4STRINGoptional

Outputs (1)

NameType
lora_pathSTRING