File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed
tensorrt_llm/_torch/models Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -979,7 +979,7 @@ def multimodal_data_device_paths(self) -> List[str]:
979979 return [
980980 "image.pixel_values" , "image.image_grid_thw" ,
981981 "video.pixel_values_videos" , "video.video_grid_thw" ,
982- "multimodal_embedding" , "mrope_config.mrope_position_ids"
982+ "multimodal_embedding"
983983 ]
984984
985985 def load_weights (self , weights , weight_mapper : BaseWeightMapper ):
@@ -1032,12 +1032,12 @@ def multimodal_data_device_paths(self) -> List[str]:
10321032 return [
10331033 "image.pixel_values" , "video.pixel_values_videos" ,
10341034 "image.image_grid_thw" , "video.video_grid_thw" ,
1035- "multimodal_embedding" , "mrope_config.mrope_position_ids"
1035+ "multimodal_embedding"
10361036 ]
10371037 else :
10381038 return [
10391039 "image.pixel_values" , "video.pixel_values_videos" ,
1040- "multimodal_embedding" , "mrope_config.mrope_position_ids"
1040+ "multimodal_embedding"
10411041 ]
10421042
10431043 def load_weights (self , weights , weight_mapper : BaseWeightMapper ):
You can’t perform that action at this time.
0 commit comments