Skip to content

Commit 458805a

Browse files
committed
fix: update model call in get_feature to use keyword argument for patch_positions
1 parent a066036 commit 458805a

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

eval_encoder/attentive_probe.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -226,7 +226,7 @@ def video_to_images(videos: torch.Tensor) -> torch.Tensor:
226226
# Stack to create patch_positions: [bs, seq_len * frame_tokens, 3]
227227
patch_positions = torch.stack([t_positions, h_positions, w_positions], dim=-1)
228228

229-
enc_out = model(videos, patch_positions, visible_index)
229+
enc_out = model(videos, patch_positions=patch_positions)
230230
if hasattr(enc_out, "last_hidden_state"):
231231
outputs = enc_out.last_hidden_state
232232
else:

0 commit comments

Comments
 (0)