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

@@ -1942,11 +1942,7 @@ def get_vision_embeddings(self, pixel_values, input_ids=None, temporal_ids=None,
         if input_ids is not None and input_ids.shape[1] == 1:
             return None
 
-        all_temporal_ids = None
-        if temporal_ids is not None:
-            all_temporal_ids = []
-            for t in temporal_ids:
-                all_temporal_ids.extend(t)
+        all_temporal_ids = [t for seq_t in temporal_ids for t in seq_t] if temporal_ids is not None else None
 
         tgt_sizes = kwargs["tgt_sizes"]
         pixel_values_list = pixel_values