diff --git a/python/sglang/multimodal_gen/runtime/layers/rotary_embedding/__init__.py b/python/sglang/multimodal_gen/runtime/layers/rotary_embedding/__init__.py index b126cf3a1f36..977c34cc348b 100644 --- a/python/sglang/multimodal_gen/runtime/layers/rotary_embedding/__init__.py +++ b/python/sglang/multimodal_gen/runtime/layers/rotary_embedding/__init__.py @@ -26,10 +26,10 @@ # limitations under the License. """Rotary Positional Embeddings — unified public API (drop-in replacement).""" -from ._base import RotaryEmbedding -from ._factory import get_rope, get_rotary_pos_embed -from ._mrope import NDRotaryEmbedding -from ._utils import ( +from .base import RotaryEmbedding +from .factory import get_rope, get_rotary_pos_embed +from .mrope import NDRotaryEmbedding +from .utils import ( _apply_rotary_emb, apply_flashinfer_rope_qk_inplace, ) diff --git a/python/sglang/multimodal_gen/runtime/layers/rotary_embedding/_base.py b/python/sglang/multimodal_gen/runtime/layers/rotary_embedding/base.py similarity index 99% rename from python/sglang/multimodal_gen/runtime/layers/rotary_embedding/_base.py rename to python/sglang/multimodal_gen/runtime/layers/rotary_embedding/base.py index 8a59dd1d1abc..6c42ca5d05bc 100644 --- a/python/sglang/multimodal_gen/runtime/layers/rotary_embedding/_base.py +++ b/python/sglang/multimodal_gen/runtime/layers/rotary_embedding/base.py @@ -4,7 +4,7 @@ from sglang.multimodal_gen.runtime.layers.custom_op import CustomOp -from ._utils import _apply_rotary_emb +from .utils import _apply_rotary_emb @CustomOp.register("rotary_embedding") diff --git a/python/sglang/multimodal_gen/runtime/layers/rotary_embedding/_factory.py b/python/sglang/multimodal_gen/runtime/layers/rotary_embedding/factory.py similarity index 97% rename from python/sglang/multimodal_gen/runtime/layers/rotary_embedding/_factory.py rename to python/sglang/multimodal_gen/runtime/layers/rotary_embedding/factory.py index f254e95eed2f..807660ea0285 100644 --- a/python/sglang/multimodal_gen/runtime/layers/rotary_embedding/_factory.py +++ b/python/sglang/multimodal_gen/runtime/layers/rotary_embedding/factory.py @@ -5,8 +5,8 @@ import torch -from ._base import LinearScalingRotaryEmbedding, RotaryEmbedding -from ._mrope import NDRotaryEmbedding, _to_tuple +from .base import LinearScalingRotaryEmbedding, RotaryEmbedding +from .mrope import NDRotaryEmbedding, _to_tuple _ROPE_DICT: dict[tuple, RotaryEmbedding] = {} _ND_ROPE_CACHE: "OrderedDict[tuple, NDRotaryEmbedding]" = OrderedDict() diff --git a/python/sglang/multimodal_gen/runtime/layers/rotary_embedding/_mrope.py b/python/sglang/multimodal_gen/runtime/layers/rotary_embedding/mrope.py similarity index 100% rename from python/sglang/multimodal_gen/runtime/layers/rotary_embedding/_mrope.py rename to python/sglang/multimodal_gen/runtime/layers/rotary_embedding/mrope.py diff --git a/python/sglang/multimodal_gen/runtime/layers/rotary_embedding/_utils.py b/python/sglang/multimodal_gen/runtime/layers/rotary_embedding/utils.py similarity index 100% rename from python/sglang/multimodal_gen/runtime/layers/rotary_embedding/_utils.py rename to python/sglang/multimodal_gen/runtime/layers/rotary_embedding/utils.py