SeedVR2-1.4B:ComfyUI対応の6レイヤー蒸留アップスケーラー

ComfyUI Wikinews

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サポートフォルダーが同梱されているため、蒸留されたアーキテクチャをそのまま読み込むことができます。

SeedVR2-1.4Bの4倍アップスケール比較(SeedVR2-3Bとの対比)

なぜ6レイヤーに蒸留するのか

SeedVR2-7Bの36個のtransformerブロックは、多くのマシンにとって重すぎます。15.35GBのfp16ウェイトと14〜16GBのピークメモリにより、一般向けノートPC、統合GPU、スマートフォンでは実行が困難です。1.4Bの生徒モデルは同じワンステップdiffusionの定式化を維持しつつ、6個のtransformerブロックのみを使用するため、常駐ウェイトとアクティベーションのワーキングセットの両方が削減されます。

SeedVR2-1.4BSeedVR2-7B(教師モデル)利点
Transformerレイヤー6366分の1
パラメータ1,442,608,2528,239,608,2325.7分の1
ディスク上のウェイト(fp16)2.69 GB15.35 GB5.7分の1
512→2048(4倍)20.2〜22.5秒33.0〜38.4秒約1.6倍高速
512→2048のピークRAM4.6 GB14.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のソースファイルを編集せずにこの問題を解決するドロップインサポートフォルダーが同梱されています。

  1. comfyui/ComfyUI-SeedVR2-1.4B/ComfyUI/custom_nodes/ にコピーして再起動
  2. ComfyUI用バリアントのチェックポイントを使用:seedvr2_distill_6L_1.4B_sharp_fp16_comfyui.safetensors(2.69 GB)。ComfyUIのローダーが必要とするpositive_conditioning / negative_conditioningテンソルが含まれています
  3. SeedVR2 VAE(ema_vae_fp16.safetensors、0.47 GB)と組み合わせる
  4. 既製のワークフローseedvr2_1.4b_upscale_image.jsonをドラッグ&ドロップ:画像を読み込む → 画像を拡大(SeedVR2 1.4B)→ 画像を保存。ワンステップ、プロンプト不要
SeedVR2-1.4B用の既製ComfyUIワークフロー

オプションの蒸留デコーダー

このリポジトリには、オプションの蒸留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でサインインしてディスカッションに参加しましょう。

コメントを読み込み中…
SeedVR2-1.4B:ComfyUI対応の6レイヤー蒸留アップスケーラー | ComfyUI Wiki