AI Toolkit Trainer

Træn WAN 2.2 og Z-Image LoRA'er med ostris AI-Toolkit: parametre, RunPod opsætning, ComfyUI brug

Af artokun · artokun/comfyui-mcp

Virker med opsætning ★ 7.2/10

AI Toolkit Trainer — Træn WAN 2.2 og Z-Image LoRA'er med ostris AI-Toolkit: parametre, RunPod opsætning, ComfyUI brug

Hvad det gør

En referencefærdighed, der vejleder i træning af brugerdefinerede LoRA'er med den standalone ostris AI-Toolkit træner, der dækker WAN 2.2/2.1 (karakter, stil, videobevægelse) og Z-Image (Turbo/Base) modeller. Den giver lokale og RunPod installationsnoter, datasæt/caption-forberedelse, startpunkt-træningsparametre, VRAM-vejledning og hvordan man indlæser den resulterende .safetensors LoRA i et ComfyUI LoraLoaderModelOnly workflow. Udløses, når brugeren ønsker at træne en WAN eller Z-Image LoRA.

Testrapport

Hentede SKILL.md via rå GitHub og klonede til en isoleret HOME temp-mappe; frontmatter parser med name+description (plus et globs-felt), og ingen curl|sh/base64/eval-lugte er til stede i færdigheden. Stikprøvekontrollerede søskendepakker, 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 (dangling references). Kunne ikke måle output: leverancen (træning af en LoRA) kræver den eksterne ostris AI-Toolkit-app, en trænings-GPU og et datasæt, ingen af disse kan køres her, så enhver artefakt ville blot gentage færdighedens egne parametertabeller. Docs er bemærkelsesværdigt ærlige, med en eksplicit Unverified-sektion, der markerer, at parametertabellerne er syntetiserede startpunkter, ikke læst fra repoets config/examples.

Testet: 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)

Kommandoer og eksempelprompter

  • /ai-toolkit-trainerTræn WAN 2.2 og Z-Image LoRA'er med ostris AI-Toolkit: parametre, RunPod opsætning, ComfyUI brug

Skills udløses af almindelige forespørgsler — ingen kommandoer at huske. Efter installationen 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