From 87ae18e2038797ea34d8eedf799220acaeaac344 Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Thu, 29 May 2025 00:14:35 +0300
Subject: [PATCH] Error on trying to use VACE with I2V models.

---
 nodes.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/nodes.py b/nodes.py
index 5a0af34..7618989 100644
--- a/nodes.py
+++ b/nodes.py
@@ -570,6 +570,8 @@ class WanVideoModelLoader:
 
         vace_layers, vace_in_dim = None, None
         if "vace_blocks.0.after_proj.weight" in sd:
+            if in_channels != 16:
+                raise ValueError("VACE only works properly with T2V models.")
             model_type = "t2v"
             if dim == 5120:
                 vace_layers = [0, 5, 10, 15, 20, 25, 30, 35]
@@ -1760,6 +1762,8 @@ class WanVideoPhantomEmbeds:
             samples = torch.cat([samples, phantom_latent_4["samples"].squeeze(0)], dim=1)
         C, T, H, W = samples.shape
 
+        log.info(f"Phantom latents shape: {samples.shape}")
+
         target_shape = (16, (num_frames - 1) // vae_stride[0] + 1 + T,
                         H * 8 // vae_stride[1],
                         W * 8 // vae_stride[2])