ComfyUI Node: Lora Training in Comfy (Advanced)

Authored by LarryJane491

Created 10 months ago

Updated 3 months ago

364 stars

Category

LJRE/LORA

Inputs

ckpt_name

v2

• No
• Yes

networkmodule

• networks.lora
• lycoris.kohya

networkdimension INT

networkalpha INT

trainingresolution INT

data_path STRING

batch_size INT

max_train_epoches INT

save_every_n_epochs INT

keeptokens INT

minSNRgamma FLOAT

learningrateText FLOAT

learningrateUnet FLOAT

learningRateScheduler

• cosine_with_restarts
• linear
• cosine
• polynomial
• constant
• constant_with_warmup

lrRestartCycles INT

optimizerType

• AdamW8bit
• Lion8bit
• SGDNesterov8bit
• AdaFactor
• prodigy

output_name STRING

algorithm

• lora
• loha
• lokr
• ia3
• dylora
• locon

networkDropout FLOAT

clip_skip INT

output_dir STRING

Outputs

Extension: Lora-Training-in-Comfy

If you see this message, your ComfyUI-Manager is outdated. Recent channel provides only the list of the latest nodes. If you want to find the complete node list, please go to the Default channel. Making LoRA has never been easier!

Authored by LarryJane491