Update nodes.py

2025-12-10 20:05:49 +02:00
parent f9287e3ecd
commit c36e56875e
1 changed files with 3 additions and 0 deletions
@@ -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)