Skip to content

Commit c8f1011

Browse files
committed
Address comments
1 parent e3a3184 commit c8f1011

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

vllm/model_executor/models/minicpmv.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -464,6 +464,8 @@ def _get_image_bounds(
464464
im_end_id: torch.Tensor,
465465
slice_start_id: Optional[torch.Tensor] = None,
466466
slice_end_id: Optional[torch.Tensor] = None) -> torch.Tensor:
467+
# All the images in the batch should share the same special image
468+
# bound token ids.
467469
start_cond = input_ids == im_start_id[0]
468470
end_cond = input_ids == im_end_id[0]
469471
if slice_start_id is not None:

0 commit comments

Comments
 (0)