AI Toolkit Trainer

Träna WAN 2.2 och Z-Image LoRA:er med ostris AI-Toolkit: parametrar, RunPod-inställning, ComfyUI-användning

Fungerar med konfiguration ★ 7.2/10

Vad den gör

En referensfärdighet som vägleder träning av anpassade LoRA:er med den fristående ostris AI-Toolkit-tränaren, som täcker WAN 2.2/2.1 (karaktär, stil, videorörelse) och Z-Image (Turbo/Base)-modeller. Den tillhandahåller lokala och RunPod-installationsanteckningar, dataset-/caption-förberedelse, startpunkts-träningsparametrar, VRAM-vägledning och hur man laddar den resulterande .safetensors LoRA i ett ComfyUI LoraLoaderModelOnly-arbetsflöde. Utlöses när användaren vill träna en WAN eller Z-Image LoRA.

Testrapport

Hämtade SKILL.md via rå GitHub och klonade till en isolerad HOME temp-katalog; frontmatter parsas med name+description (plus ett globs-fält), och inga curl|sh/base64/eval-lukter finns i färdigheten. Kontrollerade syskonpaket den refererar: wan-t2v-video och anima-lora-trainer returnerar 200, men z-image-base, z-image-turbo och z-image-xy-plot alla 404 (dangling references). Kunde inte mäta utdata: leveransen (träning av en LoRA) kräver den externa ostris AI-Toolkit-appen, en tränings-GPU och ett dataset, inget körbart här, så varje artefakt skulle bara upprepa färdighetens egna parametertabeller. Dokumentationen är anmärkningsvärt ärlig, med en explicit Unverified-sektion som flaggar att parametertabellerna är syntetiserade startpunkter, inte lästa från repots config/examples.

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

Installation

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)

Kommandon och exempelprompter

/ai-toolkit-trainerTräna WAN 2.2 och Z-Image LoRA:er med ostris AI-Toolkit: parametrar, RunPod-inställning, ComfyUI-användning

Skills triggas av vanliga förfrågningar — inga kommandon att memorera. Efter installationen aktiverar prompter som dessa skillen (på engelska):

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

Läs hela rapporten →