From 6f699b3d2d9ef1af1ef6e8e12e3d6aa57cd15568 Mon Sep 17 00:00:00 2001 From: Arthur Zucker Date: Tue, 24 Sep 2024 16:03:16 +0200 Subject: [PATCH] update --- .../models/llava_next_video/modeling_llava_next_video.py | 1 + .../models/llava_next_video/modular_llava_next_video.py | 1 + 2 files changed, 2 insertions(+) diff --git a/src/transformers/models/llava_next_video/modeling_llava_next_video.py b/src/transformers/models/llava_next_video/modeling_llava_next_video.py index b73b2f6994d9..46b9b23bd66b 100644 --- a/src/transformers/models/llava_next_video/modeling_llava_next_video.py +++ b/src/transformers/models/llava_next_video/modeling_llava_next_video.py @@ -908,6 +908,7 @@ def forward( image_features, feature_lens = self.pack_image_features( image_features, image_sizes, + self.vision_feature_select_strategy, image_newline=self.image_newline, ) diff --git a/src/transformers/models/llava_next_video/modular_llava_next_video.py b/src/transformers/models/llava_next_video/modular_llava_next_video.py index f0ec4578e488..f48056cfb97e 100644 --- a/src/transformers/models/llava_next_video/modular_llava_next_video.py +++ b/src/transformers/models/llava_next_video/modular_llava_next_video.py @@ -413,6 +413,7 @@ def forward( image_features, feature_lens = self.pack_image_features( image_features, image_sizes, + self.vision_feature_select_strategy, image_newline=self.image_newline, )