SeedVR2-1.4B: ampliador destilado de 6 capas con soporte para ComfyUI
Una destilación de 6 capas y 1.4B de SeedVR2-7B lleva la ampliación 4x/8x de un solo paso a máquinas con poca VRAM, con un nodo ComfyUI de instalación directa y workflow listo.
SeedVR2-1.4B es una destilación de conocimiento de 6 capas y 1.44B de parámetros de ByteDance-Seed/SeedVR2-7B (la variante EMA sharp). Es un ampliador de imágenes por difusión de un solo paso que reduce el modelo maestro de 8.2B parámetros / 15.35 GB a 1.44B parámetros / 2.69 GB, e incluye una carpeta de soporte completa para ComfyUI para que la arquitectura destilada se cargue sin configuración adicional.
Por qué destilar a seis capas
Los 36 bloques transformer de SeedVR2-7B son demasiado pesados para muchas máquinas: 15.35 GB de pesos fp16 y un pico de 14–16 GB lo dejan fuera del alcance de portátiles de consumo, GPU integradas y teléfonos. El modelo estudiante de 1.4B mantiene la misma formulación de difusión de un solo paso, pero utiliza solo 6 bloques transformer, lo que reduce tanto los pesos residentes como el conjunto de trabajo de activaciones.
| SeedVR2-1.4B | SeedVR2-7B (maestro) | Ventaja | |
|---|---|---|---|
| Capas de transformer | 6 | 36 | 6× menos |
| Parámetros | 1,442,608,252 | 8,239,608,232 | 5.7× menos |
| Pesos en disco (fp16) | 2.69 GB | 15.35 GB | 5.7× más pequeño |
| 512→2048 (4×) | 20.2–22.5 s | 33.0–38.4 s | ≈1.6× más rápido |
| Pico de RAM 512→2048 | 4.6 GB | 14.2–16.4 GB | ≈3× más ligero |
| 512→4096 (8×) | 54.4–55.2 s | 257–307 s | 4.7–5.6× más rápido |
Medido en un Apple M2 Ultra (64 GB) mediante MLX, con la carga del modelo incluida.
A 4×, el estudiante sigue de cerca al maestro de 36 capas en cuanto a estructura y de hecho aporta más energía de bordes: las salidas se ven nítidas y detalladas en lugar de suaves. Si tu limitación es la memoria, la latencia, o procesas muchas imágenes por lotes, este es el modelo que debes usar.
Soporte para ComfyUI (carpeta de instalación directa)
ComfyUI es compatible con SeedVR2 de forma nativa, pero identifica la arquitectura inspeccionando el checkpoint y solo reconoce blocks.31 (3B) y blocks.35 (7B). El modelo de 1.4B tiene seis bloques, por lo que no se cargará hasta que se le enseñe a ComfyUI. El repositorio incluye una carpeta de soporte de instalación directa que soluciona esto sin editar ningún archivo fuente de ComfyUI:
- Copia
comfyui/ComfyUI-SeedVR2-1.4B/enComfyUI/custom_nodes/y reinicia - Usa el checkpoint de la variante para ComfyUI:
seedvr2_distill_6L_1.4B_sharp_fp16_comfyui.safetensors(2.69 GB); incluye los tensorespositive_conditioning/negative_conditioningque requiere el cargador de ComfyUI - Combínalo con el VAE de SeedVR2 (
ema_vae_fp16.safetensors, 0.47 GB) - Arrastra el flujo de trabajo listo para usar
seedvr2_1.4b_upscale_image.json: Cargar Imagen → Ampliar Imagen (SeedVR2 1.4B) → Guardar Imagen, un solo paso, sin prompt
Decodificador destilado opcional
El repositorio también incluye un decodificador VAE destilado opcional (ema_vae_distill_fp16.safetensors, 0.19 GB) que es 112× más pequeño y ~13× más rápido de decodificar, a costa de un suave efecto de paso bajo. Puedes cambiarlo mediante la carpeta ComfyUI-SeedVR2-1.4B-TinyVAE si quieres la decodificación más rápida posible.
Disponibilidad
- Modelo: lvladikov/SeedVR2-1.4B (Apache-2.0)
- Maestro: ByteDance-Seed/SeedVR2-7B
- Pesos de referencia para ComfyUI: Comfy-Org/SeedVR2
- La ampliación no requiere prompt: no es necesario descargar ningún text encoder ni configuración, solo el transformer y el VAE
Comentarios
Inicia sesión con GitHub para unirte a la conversación.