Update nodes_sam3.py

This commit is contained in:
kijai 2026-04-15 01:28:08 +03:00
parent e3b25487a2
commit 9ad66015f7

View File

@ -429,7 +429,7 @@ class SAM3_TrackPreview(io.ComfyNode):
grid_y = torch.arange(H, device=gpu).view(1, H, 1)
grid_x = torch.arange(W, device=gpu).view(1, 1, W)
for t in range(N):
if images is not None:
if images is not None and t < images.shape[0]:
frame = images[t].clone()
else:
frame = torch.zeros(H, W, 3)