diff --git a/python/sglang/srt/layers/rotary_embedding.py b/python/sglang/srt/layers/rotary_embedding.py index 7c0a27c59..a89a8a924 100644 --- a/python/sglang/srt/layers/rotary_embedding.py +++ b/python/sglang/srt/layers/rotary_embedding.py @@ -1481,6 +1481,8 @@ class MRotaryEmbedding(RotaryEmbedding): num_tokens = positions.shape[-1] cos_sin = self.cos_sin_cache[positions] cos, sin = cos_sin.chunk(2, dim=-1) + cos = cos.contiguous() + sin = sin.contiguous() query_shape = query.shape key_shape = key.shape if positions.ndim == 2: