Skip to content

Commit b9bfb6d

Browse files
committed
fix code style
1 parent b2496c0 commit b9bfb6d

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

optimum/intel/openvino/modeling_visual_language.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1405,7 +1405,7 @@ def get_vision_embeddings(self, pixel_values, input_ids=None, **kwargs):
14051405
if input_ids is not None and input_ids.shape[1] == 1:
14061406
return None
14071407
if isinstance(pixel_values, list) or pixel_values.ndim == 5:
1408-
concat_images = torch.cat([image for image in pixel_values], dim=0)
1408+
concat_images = torch.cat(pixel_values, dim=0) if isinstance(pixel_values, list) else pixel_values
14091409
image_features = torch.from_numpy(self.vision_embeddings(concat_images).last_hidden_state)
14101410
split_sizes = [image.shape[0] for image in pixel_values]
14111411
image_features = torch.split(image_features, split_sizes, dim=0)

0 commit comments

Comments
 (0)