Update nodes.py

2025-04-02 15:29:32 +03:00
parent 8db001b586
commit 11b2c316d2
1 changed files with 2 additions and 0 deletions
@@ -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(