SeedVR2-1.4B:ComfyUI対応の6レイヤー蒸留アップスケーラー
1.4Bパラメータ、6レイヤー蒸留版のSeedVR2-7Bが、低VRAMマシンにワンステップの4倍/8倍アップスケーリングを実現。ドロップイン対応のComfyUIカスタムノードと既製ワークフロー付き。
SeedVR2-1.4B は、ByteDance-Seed/SeedVR2-7B(シャープなEMAバリアント)の1.44Bパラメータ、6レイヤー知識蒸留モデルです。ワンステップのdiffusion画像アップスケーラーであり、教師モデルを8.2Bパラメータ / 15.35GBから1.44Bパラメータ / 2.69GBへと削減し、完全なComfyUIサポートフォルダーが同梱されているため、蒸留されたアーキテクチャをそのまま読み込むことができます。
なぜ6レイヤーに蒸留するのか
SeedVR2-7Bの36個のtransformerブロックは、多くのマシンにとって重すぎます。15.35GBのfp16ウェイトと14〜16GBのピークメモリにより、一般向けノートPC、統合GPU、スマートフォンでは実行が困難です。1.4Bの生徒モデルは同じワンステップdiffusionの定式化を維持しつつ、6個のtransformerブロックのみを使用するため、常駐ウェイトとアクティベーションのワーキングセットの両方が削減されます。
| SeedVR2-1.4B | SeedVR2-7B(教師モデル) | 利点 | |
|---|---|---|---|
| Transformerレイヤー | 6 | 36 | 6分の1 |
| パラメータ | 1,442,608,252 | 8,239,608,232 | 5.7分の1 |
| ディスク上のウェイト(fp16) | 2.69 GB | 15.35 GB | 5.7分の1 |
| 512→2048(4倍) | 20.2〜22.5秒 | 33.0〜38.4秒 | 約1.6倍高速 |
| 512→2048のピークRAM | 4.6 GB | 14.2〜16.4 GB | 約3分の1の軽量さ |
| 512→4096(8倍) | 54.4〜55.2秒 | 257〜307秒 | 4.7〜5.6倍高速 |
Apple M2 Ultra(64 GB)上でMLX経由で測定。モデルの読み込み時間を含みます。
4倍拡大時、生徒モデルは構造において36レイヤーの教師モデルに密に追従し、実際にはより多くのエッジエネルギーを提供します。出力はソフトではなくシャープでディテール豊かに仕上がります。メモリ制約がある場合、レイテンシ制約がある場合、または大量の画像をバッチ処理する場合、このモデルが最適です。
ComfyUIサポート(ドロップインフォルダー)
ComfyUIはSeedVR2をネイティブにサポートしていますが、チェックポイントを検査してアーキテクチャを識別し、blocks.31(3B)とblocks.35(7B)のみを認識します。1.4Bモデルには6個のブロックがあるため、ComfyUIにそのことを認識させるまで読み込めません。このリポジトリには、ComfyUIのソースファイルを編集せずにこの問題を解決するドロップインサポートフォルダーが同梱されています。
comfyui/ComfyUI-SeedVR2-1.4B/をComfyUI/custom_nodes/にコピーして再起動- ComfyUI用バリアントのチェックポイントを使用:
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分の1、速度は約13倍高速ですが、ややローパス効果がかかります。可能な限り最速のデコードを求める場合は、ComfyUI-SeedVR2-1.4B-TinyVAEフォルダー経由で入れ替えてください。
入手方法
- モデル: lvladikov/SeedVR2-1.4B(Apache-2.0)
- 教師モデル: ByteDance-Seed/SeedVR2-7B
- ComfyUIリファレンスウェイト: Comfy-Org/SeedVR2
- アップスケーリングはプロンプト不要:テキストエンコーダーや設定ファイルのダウンロードは不要で、transformerとVAEだけで動作します
コメント
GitHubでサインインしてディスカッションに参加しましょう。