File tree Expand file tree Collapse file tree 1 file changed +7
-6
lines changed
vllm/model_executor/models Expand file tree Collapse file tree 1 file changed +7
-6
lines changed Original file line number Diff line number Diff line change @@ -457,12 +457,13 @@ def get_embedding(
457
457
458
458
return vlm_embedding , vision_hidden_states
459
459
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 :
466
467
start_cond = input_ids == im_start_id [0 ]
467
468
end_cond = input_ids == im_end_id [0 ]
468
469
if slice_start_id is not None :
You can’t perform that action at this time.
0 commit comments