Update optimum/intel/openvino/modeling_visual_language.py

Co-authored-by: Nikita Savelyev <[email protected]>

Author: openvino-dev-samples

optimum/intel/openvino/modeling_visual_language.py

@@ -287,19 +287,10 @@ def __init__(self, model: ov.Model, parent_model: OVBaseModel) -> None:
 
     def forward(self, image_feature, pos_embed, key_padding_mask, temporal_embed=None):
         self.compile()
+        inputs = {"image_feature": image_feature, "pos_embed": pos_embed, "key_padding_mask": key_padding_mask}
         if temporal_embed is not None:
-            result = self.request(
-                {
-                    "image_feature": image_feature,
-                    "pos_embed": pos_embed,
-                    "key_padding_mask": key_padding_mask,
-                    "temporal_embed": temporal_embed,
-                }
-            )[0]
-        else:
-            result = self.request(
-                {"image_feature": image_feature, "pos_embed": pos_embed, "key_padding_mask": key_padding_mask}
-            )[0]
+            inputs["temporal_embed"] = temporal_embed
+        result = self.request(inputs)[0]
         return result