File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed
vllm/model_executor/models Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -974,7 +974,7 @@ def _process_image_input(
974
974
grid_thw_list = grid_thw .tolist ()
975
975
976
976
if image_input ["type" ] == "image_embeds" :
977
- image_embeds = image_input ["image_embeds" ]
977
+ image_embeds = image_input ["image_embeds" ]. type ( self . visual . dtype )
978
978
else :
979
979
pixel_values = image_input ["pixel_values" ]
980
980
image_embeds = self .visual (pixel_values , grid_thw = grid_thw_list )
@@ -994,7 +994,7 @@ def _process_video_input(
994
994
grid_thw_list = grid_thw .tolist ()
995
995
996
996
if video_input ["type" ] == "video_embeds" :
997
- video_embeds = video_input ["video_embeds" ]
997
+ video_embeds = video_input ["video_embeds" ]. type ( self . visual . dtype )
998
998
else :
999
999
pixel_values_videos = video_input ["pixel_values_videos" ]
1000
1000
video_embeds = self .visual (pixel_values_videos ,
You can’t perform that action at this time.
0 commit comments