fix(seedvr2): detect 7B checkpoints

2026-08-31 17:21:26 +08:00
parent 1da61ed075
commit 3b1434d09f
1 changed files with 3 additions and 0 deletions
@@ -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