Fixes to mrope_config. fields

moraxu · moraxu · commit 4035427f26c2 · 2026-06-24T16:59:34.000-07:00
Signed-off-by: Michal Guzek &lt;mguzek@nvidia.com&gt;
diff --git a/tensorrt_llm/_torch/models/modeling_qwen3_5.py b/tensorrt_llm/_torch/models/modeling_qwen3_5.py
@@ -434,6 +434,8 @@ def multimodal_data_device_paths(self) -> List[str]:
             "image.pixel_values",
             "video.pixel_values_videos",
             "multimodal_embedding",
+            "mrope_config.mrope_position_ids",
+            "mrope_config.mrope_position_deltas",
         ]
 
     def load_weights(self, weights: Dict[str, torch.Tensor], weight_mapper: BaseWeightMapper):

Original file line number	Diff line number	Diff line change
`@@ -434,6 +434,8 @@ def multimodal_data_device_paths(self) -> List[str]:`
`434`	`434`	`"image.pixel_values",`
`435`	`435`	`"video.pixel_values_videos",`
`436`	`436`	`"multimodal_embedding",`
	`437`	`+ "mrope_config.mrope_position_ids",`
	`438`	`+ "mrope_config.mrope_position_deltas",`
`437`	`439`	`]`
`438`	`440`
`439`	`441`	`def load_weights(self, weights: Dict[str, torch.Tensor], weight_mapper: BaseWeightMapper):`