SeedVR2-1.4B: 支持 ComfyUI 的 6 层蒸馏图像放大模型
SeedVR2-7B 的 1.4B 参数、6 层蒸馏版本为低显存机器带来一步式 4x/8x 图像放大，附带即插即用的 ComfyUI 自定义节点和现成工作流。
SeedVR2-1.4B 是 ByteDance-Seed/SeedVR2-7B（锐化 EMA 变体）的 1.44B 参数、6 层知识蒸馏版本。它是一款一步式扩散图像放大模型，将教师模型从 8.2B 参数 / 15.35 GB 缩减至 1.44B 参数 / 2.69 GB，并附带完整的 ComfyUI 支持文件夹，使蒸馏架构开箱即可加载。
为什么要蒸馏到六层
SeedVR2-7B 的 36 个 transformer 块对许多机器来说过于沉重：15.35 GB 的 fp16 权重和 14–16 GB 的峰值内存，使其在消费级笔记本电脑、集成 GPU 和手机上无法运行。1.4B 学生模型保留了相同的一步式扩散公式，但只使用 6 个 transformer 块，同时削减了常驻权重和激活工作集。
| SeedVR2-1.4B | SeedVR2-7B（教师模型） | 优势 | |
|---|---|---|---|
| Transformer 层 | 6 | 36 | 少 6 倍 |
| 参数 | 1,442,608,252 | 8,239,608,232 | 少 5.7 倍 |
| 磁盘权重（fp16） | 2.69 GB | 15.35 GB | 小 5.7 倍 |
| 512→2048（4×） | 20.2–22.5 秒 | 33.0–38.4 秒 | 约快 1.6 倍 |
| 512→2048 峰值内存 | 4.6 GB | 14.2–16.4 GB | 约轻 3 倍 |
| 512→4096（8×） | 54.4–55.2 秒 | 257–307 秒 | 快 4.7–5.6 倍 |
在 Apple M2 Ultra（64 GB）上通过 MLX 测量，包含模型加载时间。
在 4 倍放大下，学生模型在结构上紧密追随 36 层教师模型，实际上还提供了更强的边缘能量：输出图像清晰锐利、细节丰富，而非柔和模糊。如果你受限于内存、受限于延迟，或者需要批量处理大量图像，这款模型就是你的选择。
ComfyUI 支持（即插即用文件夹）
ComfyUI 原生支持 SeedVR2，但它是通过检查 checkpoint 来识别架构的，并且只识别 blocks.31（3B）和 blocks.35（7B）。1.4B 模型只有六个块，因此在 ComfyUI 学会识别它之前无法加载。该仓库附带一个即插即用的支持文件夹，无需修改任何 ComfyUI 源文件即可解决此问题：
- 将
comfyui/ComfyUI-SeedVR2-1.4B/复制到ComfyUI/custom_nodes/并重新启动 - 使用 ComfyUI 变体 checkpoint：
seedvr2_distill_6L_1.4B_sharp_fp16_comfyui.safetensors（2.69 GB）；它包含 ComfyUI 加载器所需的positive_conditioning/negative_conditioning张量 - 搭配 SeedVR2 VAE（
ema_vae_fp16.safetensors，0.47 GB） - 拖入现成工作流
seedvr2_1.4b_upscale_image.json：加载图像 → 缩放图像（SeedVR2 1.4B）→ 保存图像，单步完成，无需提示词
可选的蒸馏解码器
该仓库还包含一个可选的蒸馏 VAE 解码器（ema_vae_distill_fp16.safetensors，0.19 GB），解码体积小 112 倍、速度快约 13 倍，代价是轻微的低通滤波效果。如果你想要尽可能快的解码速度，可以通过 ComfyUI-SeedVR2-1.4B-TinyVAE 文件夹替换使用。
可用性
- 模型： lvladikov/SeedVR2-1.4B（Apache-2.0）
- 教师模型： ByteDance-Seed/SeedVR2-7B
- ComfyUI 参考权重： Comfy-Org/SeedVR2
- 图像放大无需提示词：不需要文本编码器或配置文件，只需要 transformer 和 VAE
评论
使用 GitHub 登录后即可参与讨论。