SeedVR2-1.4B : upscaler distillé 6 couches avec support ComfyUI

SeedVR2-1.4B est une distillation de connaissances en 6 couches de 1,44 milliard de paramètres de ByteDance-Seed/SeedVR2-7B (la variante EMA sharp). Il s'agit d'un upscaler d'image à diffusion en une étape qui réduit le modèle enseignant de 8,2 milliards de paramètres / 15,35 Go à 1,44 milliard de paramètres / 2,69 Go, avec un dossier de prise en charge ComfyUI complet pour que l'architecture distillée se charge directement.

Pourquoi distiller en six couches

Les 36 blocs transformer de SeedVR2-7B sont trop lourds pour de nombreuses machines : 15,35 Go de poids fp16 et un pic de 14 à 16 Go le rendent inaccessible sur les ordinateurs portables grand public, les GPU intégrés et les téléphones. Le modèle étudiant 1.4B conserve la même formulation de diffusion en une étape mais n'utilise que 6 blocs transformer, ce qui réduit à la fois les poids résidents et l'ensemble de travail d'activation.

Mesuré sur un Apple M2 Ultra (64 Go) via MLX, chargement du modèle inclus.

En 4×, l'étudiant suit de près le modèle enseignant de 36 couches sur le plan structurel et fournit en réalité davantage d'énergie de contour : les sorties sont nettes et détaillées plutôt que douces. Si vous êtes limité par la mémoire, par la latence ou si vous traitez beaucoup d'images par lots, c'est le modèle qu'il vous faut.

Prise en charge de ComfyUI (dossier prêt à l'emploi)

ComfyUI prend en charge SeedVR2 nativement, mais il identifie l'architecture en inspectant le checkpoint et ne reconnaît que blocks.31 (3B) et blocks.35 (7B). Le modèle 1.4B possède six blocs, il ne se chargera donc pas tant que ComfyUI n'aura pas été informé de son existence. Le dépôt fournit un dossier de prise en charge prêt à l'emploi qui corrige ce problème sans modifier aucun fichier source de ComfyUI :

1. Copiez comfyui/ComfyUI-SeedVR2-1.4B/ dans ComfyUI/custom_nodes/ puis redémarrez
2. Utilisez le checkpoint variante ComfyUI : seedvr2_distill_6L_1.4B_sharp_fp16_comfyui.safetensors (2,69 Go) ; il contient les tenseurs positive_conditioning / negative_conditioning requis par le chargeur de ComfyUI
3. Associez-le au VAE SeedVR2 (ema_vae_fp16.safetensors, 0,47 Go)
4. Glissez-déposez le flux de travail prêt à l'emploi seedvr2_1.4b_upscale_image.json : Charger Image → Agrandir l'image (SeedVR2 1.4B) → Enregistrer l'image, en une étape, sans prompt

Télécharger le flux de travail de mise à l'échelle SeedVR2-1.4B

Décodeur distillé optionnel

Le dépôt inclut également un décodeur VAE distillé optionnel (ema_vae_distill_fp16.safetensors, 0,19 Go) qui est **112×