We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 38168fb commit 901a5f4Copy full SHA for 901a5f4
vllm/model_executor/models/molmo.py
@@ -1520,13 +1520,12 @@ def _process_image_input(
1520
image_features_flat = self.vision_backbone(
1521
images=images_flat.unsqueeze(0),
1522
image_masks=(None if image_masks_flat is None else
1523
- image_masks_flat.unsqueeze(0)),
+ image_masks_flat.unsqueeze(0)),
1524
).squeeze(0)
1525
1526
# Only the features corresponding to patch tokens are relevant
1527
return [
1528
- feats[f_is_patch]
1529
- for feats, f_is_patch in zip(
+ feats[f_is_patch] for feats, f_is_patch in zip(
1530
image_features_flat.split(num_crops.tolist()),
1531
feat_is_patch_flat.split(num_crops.tolist()),
1532
)
0 commit comments