Kroma v0.3.1: Destilación on-policy para el fine-tune de Krea 2

Qué cambia OPD

Los checkpoints turbo anteriores de Kroma se destilaban offline: el estudiante se entrenaba para imitar al teacher en un calendario de muestreo fijo. El readme describe directamente el modo de fallo de esa receta. Imitar al teacher en estados que el teacher eligió arrastra poco a poco al estudiante fuera del manifold de datos original, lo que se manifiesta como colapso de modos, detalles desvaídos y prompts que de repente dejan de funcionar.

OPD, la destilación on-policy, invierte la dirección. El estudiante genera sus propias trayectorias y el teacher lo corrige en esos puntos exactos. El entrenamiento solo ocurre en estados que el modelo visita realmente, así que el checkpoint destilado se mantiene dentro de la distribución del modelo del que proviene. El resultado pretende ser la velocidad de Turbo sin el coste habitual de la destilación.

Qué hay ahora en el repositorio

El repositorio se reorganizó el 7 de octubre y ahora contiene tres archivos:

Los archivos kroma-v0.2-turbo, kroma-v0.2-base, kroma-v0.3-turbo y kroma-v0.3-tdm-4steps-artifact a los que enlazaban artículos anteriores de Kroma han sido eliminados, por lo que las instrucciones de descarga antiguas ya no resuelven. La tabla "files to download" del readme todavía nombra kroma-v0.2-turbo.safetensors, lo cual está obsoleto: v0.3.1 OPD es el archivo que hay que descargar.

Cómo ejecutarlo en ComfyUI

Ejecutar v0.3.1 es idéntico a v0.2 y v0.3. Usa el stack estándar de Krea 2, así que no se necesita ningún nodo personalizado más allá del soporte nativo de Krea 2 en ComfyUI:

1. Coloca kroma-v0.3.1-turbo-opd.safetensors en ComfyUI/models/diffusion_models/
2. Carga el text encoder de Krea 2 (Qwen3-VL, 12 capas) con un CLIPLoader configurado al tipo krea2
3. Carga el VAE de Krea 2
4. Muestrea con las configuraciones Turbo para las que se destiló el checkpoint: 8-12 steps, CFG 1.0-1.5, shift (mu) 1.15

La cadena mínima es un Cargar Modelo de Difusión conectado a un KSampler, con el CLIP krea2 y el VAE de Krea 2 alimentándolo. Si la salida no se ve bien, el primer sospechoso del readme es el text encoder: confirma que el tipo del CLIPLoader es krea2 y que ComfyUI está en una build actual con soporte nativo de Krea 2.

El entrenamiento de LoRA sigue funcionando

Como OPD mantiene al estudiante dentro de la distribución original, los LoRA entrenados de la forma habitual siguen funcionando. Entrena con el teacher (kroma-sensei-booru-e6-teacher-velocity-v0.3.safetensors) o con la base (kroma-v0.3-base.safetensors), y el LoRA resultante se carga en kroma-v0.3.1-turbo-opd.safetensors sin cambios. Esa es la principal diferencia práctica frente a un estudiante destilado de forma agresiva: no tienes que reentrenar ni volver a derivar adaptadores cuando se cambia el checkpoint turbo.

Disponibilidad

Modelo: lodestones/Kroma
Checkpoint: kroma-v0.3.1-turbo-opd.safetensors
Modelo base: Krea 2