From 11b2c316d2a046854487f0f4b5e9939021f8475f Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Wed, 2 Apr 2025 15:29:32 +0300
Subject: [PATCH] Update nodes.py

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

diff --git a/nodes.py b/nodes.py
index cde7a0e..92f7794 100644
--- a/nodes.py
+++ b/nodes.py
@@ -2614,6 +2614,8 @@ class WanVideoSampler:
                     partial_vace_context = None
                     if vace_context is not None:
                         partial_vace_context = vace_context[0][:, c, :, :]
+                        if has_ref:
+                            partial_vace_context[:, 0, :, :] = vace_context[0][:, 0, :, :]
                     partial_latent_model_input = latent_model_input[:, c, :, :]
 
                     noise_pred_context, new_teacache = predict_with_cfg(