mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-05-02 13:22:32 +08:00
Update nodes_sam3.py
This commit is contained in:
parent
e3b25487a2
commit
9ad66015f7
@ -429,7 +429,7 @@ class SAM3_TrackPreview(io.ComfyNode):
|
|||||||
grid_y = torch.arange(H, device=gpu).view(1, H, 1)
|
grid_y = torch.arange(H, device=gpu).view(1, H, 1)
|
||||||
grid_x = torch.arange(W, device=gpu).view(1, 1, W)
|
grid_x = torch.arange(W, device=gpu).view(1, 1, W)
|
||||||
for t in range(N):
|
for t in range(N):
|
||||||
if images is not None:
|
if images is not None and t < images.shape[0]:
|
||||||
frame = images[t].clone()
|
frame = images[t].clone()
|
||||||
else:
|
else:
|
||||||
frame = torch.zeros(H, W, 3)
|
frame = torch.zeros(H, W, 3)
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user