kvcache-ai/sglang
1
0
Fork 0
mirror of https://github.com/kvcache-ai/sglang.git synced 2026-07-01 04:08:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix nightly grok failure on rotary embedding import (#19192)

Browse Source 
Co-authored-by: michaelzhang-ai <michaelzhang-ai@users.noreply.github.com>
This commit is contained in:
Michael
2026-02-24 21:25:16 -08:00
committed by GitHub
parent 3fe93b5493
commit ca09d71cf0
1 changed files with 12 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
python/sglang/srt/layers/rotary_embedding/__init__.py
View File

@@ -8,6 +8,15 @@ from sglang.srt.layers.rotary_embedding.mrope import (
MRotaryEmbedding,
)
from sglang.srt.layers.rotary_embedding.utils import apply_rotary_pos_emb
from sglang.srt.layers.rotary_embedding.yarn import (
yarn_find_correction_range,
yarn_get_mscale_simple,
yarn_linear_ramp_mask,
)
_yarn_find_correction_range = yarn_find_correction_range
_yarn_get_mscale = yarn_get_mscale_simple
_yarn_linear_ramp_mask = yarn_linear_ramp_mask
__all__ = [
"RotaryEmbedding",
@@ -16,4 +25,7 @@ __all__ = [
"MRotaryEmbedding",
"Ernie4_5_VLRotaryEmbedding",
"apply_rotary_pos_emb",
"_yarn_find_correction_range",
"_yarn_get_mscale",
"_yarn_linear_ramp_mask",
]