From 15272e61d6676f0efa4247c285a345d6a78416b5 Mon Sep 17 00:00:00 2001 From: Hamish Scott Date: Thu, 18 Sep 2025 10:15:07 +0100 Subject: [PATCH] use skip_predictor in vjepa2 `get_vision_features` --- src/transformers/models/vjepa2/modeling_vjepa2.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/transformers/models/vjepa2/modeling_vjepa2.py b/src/transformers/models/vjepa2/modeling_vjepa2.py index bde505b4ea54..eedc94b845a4 100644 --- a/src/transformers/models/vjepa2/modeling_vjepa2.py +++ b/src/transformers/models/vjepa2/modeling_vjepa2.py @@ -1125,7 +1125,7 @@ def forward( return encoder_output def get_vision_features(self, pixel_values_videos) -> torch.Tensor: - encoder_output = self.forward(pixel_values_videos) + encoder_output = self.forward(pixel_values_videos, skip_predictor=True) return encoder_output.last_hidden_state