restore image_bound input filling

huggingface · Oct 15, 2024 · 24414cd · 24414cd
1 parent 68dd3ff
commit 24414cd
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/optimum/intel/openvino/modeling_visual_language.py b/optimum/intel/openvino/modeling_visual_language.py
@@ -603,8 +603,8 @@ def forward(
             attention_mask=attention_mask,
             position_ids=position_ids,
             past_key_values=past_key_values,
-            image_bound=None,
-            tgt_sizes=None,
+            image_bound=image_bound,
+            tgt_sizes=tgt_sizes,
             **kwargs,
         )
         return self.language_model.forward(