{ "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 } }