AI Toolkit Trainer

Tren WAN 2.2 og Z-Image LoRA-er med ostris AI-Toolkit: parametere, RunPod-oppsett, ComfyUI-bruk

av artokun · artokun/comfyui-mcp

Krever oppsett ★ 7.2/10

AI Toolkit Trainer — Tren WAN 2.2 og Z-Image LoRA-er med ostris AI-Toolkit: parametere, RunPod-oppsett, ComfyUI-bruk

Hva den gjør

En referanseferdighet som veileder i trening av tilpassede LoRA-er med den frittstående ostris AI-Toolkit-treneren, som dekker WAN 2.2/2.1 (karakter, stil, videobevegelse) og Z-Image (Turbo/Base) modeller. Den gir lokale og RunPod-installasjonsnotater, datasett/bildetekstforberedelse, startpunkt-treningsparametere, VRAM-veiledning, og hvordan man laster den resulterende .safetensors LoRA-en i en ComfyUI LoraLoaderModelOnly-arbeidsflyt. Utløses når brukeren ønsker å trene en WAN eller Z-Image LoRA.

Testrapport

Hentet SKILL.md via rå GitHub og klonet inn i en isolert HOME temp-katalog; frontmatter parses med name+description (pluss et globs-felt), og ingen curl|sh/base64/eval-lukter er til stede i ferdigheten. Stikkprøvekontrollerte søskenpakkene den refererer til: wan-t2v-video og anima-lora-trainer returnerer 200, men z-image-base, z-image-turbo og z-image-xy-plot alle 404 (hengende referanser). Kunne ikke måle utdata: leveransen (trening av en LoRA) krever den eksterne ostris AI-Toolkit-appen, en trenings-GPU og et datasett, ingen av disse kan kjøres her, så enhver artefakt ville bare gjenta ferdighetens egne parametertabeller. Dokumentasjonen er bemerkelsesverdig ærlig, med en eksplisitt Ubekreftet-seksjon som flagger at parametertabellene er syntetiserte startpunkter, ikke lest fra repoets config/examples.

Testet på: 2026-07-31 · Claude Code 2.x (agent harness)

Installer

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)

Kommandoer og eksempelprompter

  • /ai-toolkit-trainerTren WAN 2.2 og Z-Image LoRA-er med ostris AI-Toolkit: parametere, RunPod-oppsett, ComfyUI-bruk

Skills utløses av vanlige forespørsler — ingen kommandoer å huske. Etter installasjonen aktiverer prompter som disse skillen (på engelsk):

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