Documentation
README
AI-Toolkit LoRA Trainer (WAN 2.2 & Z-Image)
Overview
AI-Toolkit by ostris is "the ultimate training toolkit for finetuning diffusion models" (MIT license) — a standalone trainer with its own web UI, NOT a ComfyUI custom node. It runs a Node.js UI front end over a Python (run.py) training backend, and trains LoRAs for many model families — here we cover WAN 2.2 / 2.1 video models and Z-Image (Turbo & Base).
- Repo:
https://github.com/ostris/ai-toolkit(cloned by the installers). - Backend:
python run.py config/<job>.yml. UI: a Node.js app underui/that schedules/monitors jobs (you don't have to keep the UI open while a job runs). - Output: a standard
.safetensorsLoRA you drop into ComfyUImodels/loras/and load withLoraLoaderModelOnly.
This is the opening of the README. Read the full README on GitHub.