From c36e56875e993ff984a08a6a37bd322ce4ca8a63 Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Wed, 10 Dec 2025 20:05:49 +0200
Subject: [PATCH] Update nodes.py

---
 WanMove/nodes.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/WanMove/nodes.py b/WanMove/nodes.py
index df5e7b0..478b114 100644
--- a/WanMove/nodes.py
+++ b/WanMove/nodes.py
@@ -37,6 +37,9 @@ class WanVideoWanDrawWanMoveTracks:
         track = tracks["tracks"].unsqueeze(0)
         track_visibility = tracks["track_visibility"].unsqueeze(0)
         images_in = images * 255.0
+        if images_in.shape[0] != track.shape[1]:
+            repeat_count = track.shape[1] // images.shape[0]
+            images_in = images_in.repeat(repeat_count, 1, 1, 1)
         track_video = draw_tracks_on_video(images_in, track, track_visibility, track_frame=line_resolution, circle_size=circle_size, opacity=opacity, line_width=line_width)
         track_video = torch.stack([TF.to_tensor(frame) for frame in track_video], dim=0).movedim(1, -1)