AI Toolkit Trainer

Entraîner des LoRA WAN 2.2 et Z-Image avec ostris AI-Toolkit : paramètres, configuration RunPod, utilisation ComfyUI

Fonctionne avec configuration ★ 7.2/10

Ce que fait

Une compétence de référence qui guide l'entraînement de LoRA personnalisés avec l'outil d'entraînement autonome ostris AI-Toolkit, couvrant les modèles WAN 2.2/2.1 (personnage, style, mouvement vidéo) et Z-Image (Turbo/Base). Elle fournit des notes d'installation locales et RunPod, la préparation des datasets/captions, les paramètres d'entraînement de départ, des conseils sur la VRAM, et comment charger le LoRA .safetensors résultant dans un workflow ComfyUI LoraLoaderModelOnly. Se déclenche lorsque l'utilisateur souhaite entraîner un LoRA WAN ou Z-Image.

Rapport de test

SKILL.md récupéré via raw GitHub et cloné dans un répertoire temporaire HOME isolé ; le frontmatter est parsé avec name+description (plus un champ globs), et aucune trace de curl|sh/base64/eval n'est présente dans la compétence. Vérifié les packs frères qu'il référence : wan-t2v-video et anima-lora-trainer renvoient 200, mais z-image-base, z-image-turbo et z-image-xy-plot renvoient tous 404 (références orphelines). Impossible de mesurer la sortie : le livrable (entraînement d'un LoRA) nécessite l'application externe ostris AI-Toolkit, un GPU d'entraînement et un dataset, aucun n'étant exécutable ici, donc tout artefact ne ferait que réitérer les propres tables de paramètres de la compétence. La documentation est remarquablement honnête, avec une section explicite Non vérifié signalant que les tables de paramètres sont des points de départ synthétisés, non lus à partir des config/examples du dépôt.

Testé le: 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)

Commandes et exemples de prompts

/ai-toolkit-trainerEntraîner des LoRA WAN 2.2 et Z-Image avec ostris AI-Toolkit : paramètres, configuration RunPod, utilisation ComfyUI

Les skills se déclenchent sur des demandes en langage courant — aucune commande à retenir. Après installation, des prompts comme ceux-ci l'activent (en anglais) :

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

Rapport complet →