From ca09d71cf0b677bd7637c3d2ff755d2fd0aba58c Mon Sep 17 00:00:00 2001 From: Michael <13900043+michaelzhang-ai@users.noreply.github.com> Date: Tue, 24 Feb 2026 21:25:16 -0800 Subject: [PATCH] Fix nightly grok failure on rotary embedding import (#19192) Co-authored-by: michaelzhang-ai --- .../sglang/srt/layers/rotary_embedding/__init__.py | 12 ++++++++++++ 1 file changed, 12 insertions(+) diff --git a/python/sglang/srt/layers/rotary_embedding/__init__.py b/python/sglang/srt/layers/rotary_embedding/__init__.py index df1a95059..97890ef96 100644 --- a/python/sglang/srt/layers/rotary_embedding/__init__.py +++ b/python/sglang/srt/layers/rotary_embedding/__init__.py @@ -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", ]