Skip to content

Commit 901a5f4

Browse files
committed
Pass pre-commit
Signed-off-by: DarkLight1337 <tlleungac@connect.ust.hk>
1 parent 38168fb commit 901a5f4

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

vllm/model_executor/models/molmo.py

+2-3
Original file line numberDiff line numberDiff line change
@@ -1520,13 +1520,12 @@ def _process_image_input(
15201520
image_features_flat = self.vision_backbone(
15211521
images=images_flat.unsqueeze(0),
15221522
image_masks=(None if image_masks_flat is None else
1523-
image_masks_flat.unsqueeze(0)),
1523+
image_masks_flat.unsqueeze(0)),
15241524
).squeeze(0)
15251525

15261526
# Only the features corresponding to patch tokens are relevant
15271527
return [
1528-
feats[f_is_patch]
1529-
for feats, f_is_patch in zip(
1528+
feats[f_is_patch] for feats, f_is_patch in zip(
15301529
image_features_flat.split(num_crops.tolist()),
15311530
feat_is_patch_flat.split(num_crops.tolist()),
15321531
)

0 commit comments

Comments
 (0)