From 3b1434d09f139fd4222dc59fd973fe9ba6f58348 Mon Sep 17 00:00:00 2001
From: gaclove <peng.gaoc@gmail.com>
Date: Tue, 9 Jun 2026 19:49:22 +0800
Subject: [PATCH] fix(seedvr2): detect 7B checkpoints

---
 nodes/seedvr.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/nodes/seedvr.py b/nodes/seedvr.py
index 945bf2f..bf2de7a 100644
--- a/nodes/seedvr.py
+++ b/nodes/seedvr.py
@@ -351,6 +351,9 @@ class LightX2VSeedVR2Loader:
             "vae_memory_limit_gb": float(vae_memory_limit_gb),
             "cpu_offload": bool(cpu_offload),
         }
+        if "7b" in ckpt_name.lower():
+            config["model_size"] = "7b"
+
         if precision.startswith("fp8-"):
             config["dit_quantized_ckpt"] = str(ckpt_path)
             config["dit_quant_scheme"] = precision