ldc-workflow-krea2-lora / workflow.json
dev-stack lane workflow-dev
Krea 2 LoRA Trainer v1
b9fd757
Raw History Blame Contribute Delete
2.62 kB
{
"schema": 1,
"kind": "training-job",
"id": "krea2-lora",
"profile": "krea2_lora_train",
"name": "Krea 2 LoRA Trainer",
"description": "Train a Krea 2 LoRA with ostris/ai-toolkit on a rented GPU, from a validated dataset and a trigger word, then smoke-test the result against Krea 2 text-to-image. Not a ComfyUI graph: job.json holds the hyperparameters, the ai-toolkit config template and the validation policy.",
"version": {
"tag": "v1",
"ordinal": 1,
"date": "2026-10-26",
"release_note": "The trainer every run before packaging used: rank 16, AdamW 8-bit at 1e-4, 1024px, Raw or Turbo (through the training adapter) on ai-toolkit 0.13.23, judged by a krea2-txt2img v8 smoke test."
},
"generation_settings": [
{
"id": "dataset",
"displayName": "Dataset",
"description": "A .tar.gz of images, each with an optional same-named .txt caption, uploaded or fetched from an https link. It is validated before it is accepted; the manifest shows what will train.",
"control": "dataset",
"default": "",
"group": "Dataset",
"required": true
},
{
"id": "trigger_word",
"displayName": "Trigger word",
"description": "One rare token, like krx2person, spliced into every caption so the LoRA learns to answer to it. Use the same word in prompts at inference.",
"control": "text",
"default": "",
"group": "Dataset",
"required": true
},
{
"id": "checkpoint",
"displayName": "Base checkpoint",
"description": "The Krea 2 model the LoRA is trained against; its bf16 build is what trains. Empty uses Krea 2 Raw, the base intended for fine-tuning \u2014 a LoRA trained on Raw also works on Turbo. Turbo bases train through ostris's training adapter.",
"control": "checkpoint",
"default": "",
"group": "Model",
"required": false
},
{
"id": "steps",
"displayName": "Steps",
"description": "Training steps. 0 scales them to the dataset (50 per image, 1,000-3,000). More is not better past the point the samples stop improving; the operator ceiling refuses anything above it.",
"control": "number",
"default": 0,
"group": "Training",
"required": false,
"summaryLabel": "Steps",
"min": 0,
"max": 20000,
"step": 50
}
],
"media_modes": [
"krea2_lora_train"
],
"runtime": {
"image": "ostris/aitoolkit:0.13.23",
"entrypoint": [
"python",
"run.py"
],
"min_vram_gb": 48,
"hf_home": "/workspace/hf-cache",
"stall_timeout_seconds": 1800.0
}
}