AI Toolkit Trainer

Train WAN 2.2 en Z-Image LoRA's met ostris AI-Toolkit: params, RunPod setup, ComfyUI gebruik

Werkt met setup ★ 7.2/10

Wat het doet

Een referentie skill die het trainen van aangepaste LoRA's begeleidt met de standalone ostris AI-Toolkit trainer, die WAN 2.2/2.1 (character, style, video-motion) en Z-Image (Turbo/Base) modellen behandelt. Het biedt lokale en RunPod installatie-notities, dataset/caption voorbereiding, startpunt trainingsparameters, VRAM-begeleiding, en hoe de resulterende .safetensors LoRA te laden in een ComfyUI LoraLoaderModelOnly workflow. Triggert wanneer de gebruiker een WAN of Z-Image LoRA wil trainen.

Testrapport

SKILL.md opgehaald via raw GitHub en gekloond naar een geïsoleerde HOME temp dir; frontmatter parseert met name+description (plus een globs veld), en er zijn geen curl|sh/base64/eval geuren aanwezig in de skill. Steekproefsgewijs gecontroleerde sibling packs waarnaar het verwijst: wan-t2v-video en anima-lora-trainer retourneren 200, maar z-image-base, z-image-turbo en z-image-xy-plot allemaal 404 (dangling references). Kon de output niet meten: de oplevering (het trainen van een LoRA) vereist de externe ostris AI-Toolkit app, een training GPU, en een dataset, geen hiervan hier uitvoerbaar, dus elk artefact zou alleen de eigen param tabellen van de skill herhalen. Docs zijn opmerkelijk eerlijk, met een expliciete Unverified sectie die aangeeft dat de param tabellen gesynthetiseerde startpunten zijn, niet gelezen uit de config/examples van de repo.

Getest op: 2026-07-31 · Claude Code 2.x (agent harness)

Installatie

git clone --depth 1 https://github.com/artokun/comfyui-mcp.git /tmp/ai-toolkit-trainer-src
mkdir -p ~/.claude/skills
cp -R /tmp/ai-toolkit-trainer-src/plugin/skills/ai-toolkit-trainer ~/.claude/skills/ai-toolkit-trainer
# Self-contained knowledge skill: one SKILL.md, no local scripts or reference files.
# To actually TRAIN you must separately install ostris AI-Toolkit (https://github.com/ostris/ai-toolkit)
#   — a Node UI over a Python run.py backend — plus a CUDA training GPU and a captioned dataset.
#   The repo's Windows installer .bat files download launcher scripts from a third-party
#   HuggingFace repo (Aitrepreneur/FLX); review before running on a sensitive machine.
# Plugin-marketplace alternative (installs the whole comfyui-mcp plugin, 35 skills):
#   /plugin marketplace add artokun/comfyui-mcp  (repo ships .claude-plugin/marketplace.json)

Commando's en voorbeeldprompts

/ai-toolkit-trainerTrain WAN 2.2 en Z-Image LoRA's met ostris AI-Toolkit: params, RunPod setup, ComfyUI gebruik

Skills reageren op gewone verzoeken — geen commando's om te onthouden. Na installatie activeren prompts zoals deze de skill (in het Engels):

Train a WAN LoRA on this video dataset
Set up a Z-Image LoRA on RunPod
Prepare my dataset for LoRA training

Volledig rapport (Engels) →