Tooluck logoTooluck
SKILL.md

ai-toolkit-trainer

Agent skill fromartokun/comfyui-mcp

Train custom LoRAs with ostris AI-Toolkit. Covers WAN 2.2/2.1 (people, styles, video motion) and Z-Image (Turbo & Base, low-VRAM image LoRAs). Use when the user wants to train a WAN or Z-Image LoRA; covers local + RunPod setup, dataset prep, key params, and using the result in a ComfyUI workflow.

How to install ai-toolkit-trainer

install ai-toolkit-trainer
# install the comfyui-mcp pack with the skills CLI
npx skills add artokun/comfyui-mcp

# or add only this skill to Claude Code: copy the folder

plugin/skills/ai-toolkit-trainer/

# into

~/.claude/skills/ai-toolkit-trainer/ (all projects)

.claude/skills/ai-toolkit-trainer/ (one project)

plugin/skills/ai-toolkit-trainer/SKILL.md

MITFull file
1# AI-Toolkit LoRA Trainer (WAN 2.2 & Z-Image)
2
3## Overview
4
5AI-Toolkit by ostris is an MIT-licensed trainer for finetuning diffusion models. It is 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. This skill covers the WAN 2.2 / 2.1 video models and Z-Image (Turbo & Base).
6
7- Repo: `https://github.com/ostris/ai-toolkit` (cloned by the installers).
8- Backend: `python run.py config/<job>.yml`. UI: a Node.js app under `ui/` that schedules and monitors jobs. You do not have to keep the UI open while a job runs.
9- Output: a standard `.safetensors` LoRA you drop into ComfyUI `models/loras/` and load with `LoraLoaderModelOnly`.
10
11Best for:
12- WAN LoRAs. A person or character, an art style, or a specific camera or video motion, trained from image or video clip datasets. For *using* WAN see wan-t2v-video / wan-flf-video.
13- Z-Image LoRAs. Fast, very low-VRAM image LoRAs (faces, characters, outfits, styles) on the 6B Z-Image base/turbo. For *using* Z-Image see z-image-base / z-image-turbo, and the z-image-xy-plot pack to compare trained LoRAs.
14
15For low-VRAM anime image LoRAs on a different stack (kohya `sd-scripts`), see the sibling anima-lora-trainer.
16…

Excerpt shown under the repository's MIT license.

More skills in comfyui-mcp

Keep exploring

Similar skill repositories

thedotmack /

claude-mem
116k

Persistent Context Across Sessions for Every Agent – Captures everything your agent does during sessions, compresses it with AI, and injects relevant context back into future sessions. Works with Claude Code, OpenClaw, Codex, Gemini, Hermes, Copilot, OpenCode + More

CCCXGHGM+135 skills

Newsletter

Join the Community

Subscribe to our newsletter for the latest news and updates