diff --git a/python/sglang/srt/layers/rotary_embedding/mrope.py b/python/sglang/srt/layers/rotary_embedding/mrope.py index 575e34a12..979b9741b 100644 --- a/python/sglang/srt/layers/rotary_embedding/mrope.py +++ b/python/sglang/srt/layers/rotary_embedding/mrope.py @@ -303,6 +303,7 @@ class MRotaryEmbedding(RotaryEmbedding): fused_set_kv_buffer_arg=None, ) -> Tuple[torch.Tensor, torch.Tensor]: assert positions.ndim in (1, 2) + self._match_cos_sin_cache_dtype(query) if positions.ndim == 2 and self.mrope_section: multimodal_rotary_embedding( query,