Skip to content

Commit e3a3184

Browse files
committed
fix format
1 parent 7307fed commit e3a3184

File tree

1 file changed

+7
-6
lines changed

1 file changed

+7
-6
lines changed

vllm/model_executor/models/minicpmv.py

Lines changed: 7 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -457,12 +457,13 @@ def get_embedding(
457457

458458
return vlm_embedding, vision_hidden_states
459459

460-
def _get_image_bounds(self,
461-
input_ids: torch.Tensor,
462-
im_start_id: torch.Tensor,
463-
im_end_id: torch.Tensor,
464-
slice_start_id: Optional[torch.Tensor] = None,
465-
slice_end_id: Optional[torch.Tensor] = None) -> torch.Tensor:
460+
def _get_image_bounds(
461+
self,
462+
input_ids: torch.Tensor,
463+
im_start_id: torch.Tensor,
464+
im_end_id: torch.Tensor,
465+
slice_start_id: Optional[torch.Tensor] = None,
466+
slice_end_id: Optional[torch.Tensor] = None) -> torch.Tensor:
466467
start_cond = input_ids == im_start_id[0]
467468
end_cond = input_ids == im_end_id[0]
468469
if slice_start_id is not None:

0 commit comments

Comments
 (0)