AI Toolkit Trainer

Entrena LoRAs WAN 2.2 y Z-Image con ostris AI-Toolkit: parámetros, configuración RunPod, uso de ComfyUI

Funciona con configuración ★ 7.2/10

Qué hace

Una habilidad de referencia que guía el entrenamiento de LoRAs personalizadas con el entrenador autónomo ostris AI-Toolkit, cubriendo los modelos WAN 2.2/2.1 (personaje, estilo, movimiento de video) y Z-Image (Turbo/Base). Proporciona notas de instalación local y RunPod, preparación de conjuntos de datos/subtítulos, parámetros de entrenamiento de punto de partida, guía de VRAM y cómo cargar el LoRA .safetensors resultante en un flujo de trabajo ComfyUI LoraLoaderModelOnly. Se activa cuando el usuario quiere entrenar un LoRA WAN o Z-Image.

Informe de la prueba

SKILL.md obtenido a través de raw GitHub y clonado en un directorio temporal HOME aislado; el frontmatter se analiza con name+description (más un campo globs), y no hay problemas de curl|sh/base64/eval presentes en la habilidad. Se verificaron los paquetes hermanos que referencia: wan-t2v-video y anima-lora-trainer devuelven 200, pero z-image-base, z-image-turbo y z-image-xy-plot todos 404 (referencias colgantes). No se pudo medir la salida: el entregable (entrenar un LoRA) necesita la aplicación externa ostris AI-Toolkit, una GPU de entrenamiento y un conjunto de datos, ninguno ejecutable aquí, por lo que cualquier artefacto solo reafirmaría las propias tablas de parámetros de la habilidad. La documentación es notablemente honesta, con una sección explícita Unverified que señala que las tablas de parámetros son puntos de partida sintetizados, no leídos de la configuración/ejemplos del repositorio.

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

Instalación

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)

Comandos y prompts de ejemplo

/ai-toolkit-trainerEntrena LoRAs WAN 2.2 y Z-Image con ostris AI-Toolkit: parámetros, configuración RunPod, uso de ComfyUI

Los skills se activan con peticiones en lenguaje natural, sin comandos que memorizar. Tras instalarlo, prompts como estos lo activan (en inglés):

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

Informe completo →