From 18aa47cc74534241835d2682bc5b51293082529c Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Sat, 19 Apr 2025 22:41:50 +0300
Subject: [PATCH] Update nodes.py

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

diff --git a/nodes.py b/nodes.py
index c5c8c2c..3e15b0e 100644
--- a/nodes.py
+++ b/nodes.py
@@ -3018,7 +3018,7 @@ class WanVideoSampler:
 
                     partial_unianim_data = None
                     if unianim_data is not None:
-                        partial_dwpose = dwpose_data[:, c, :, :]
+                        partial_dwpose = dwpose_data[:, :, c, :, :]
                         partial_dwpose_flat=rearrange(partial_dwpose, 'b c f h w -> b (f h w) c')
                         partial_unianim_data = {
                             "dwpose": partial_dwpose_flat,