[Diffusion] Match rotary_embedding module name style (#19179)

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 .base import RotaryEmbedding
-from ._factory import get_rope, get_rotary_pos_embed
+from .factory import get_rope, get_rotary_pos_embed
-from ._mrope import NDRotaryEmbedding
+from .mrope import NDRotaryEmbedding
-from ._utils import (
+from .utils import (
     _apply_rotary_emb,
     apply_flashinfer_rope_qk_inplace,
 )

python/sglang/multimodal_gen/runtime/layers/rotary_embedding/base.py

@@ -4,7 +4,7 @@ import torch
 
 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")

python/sglang/multimodal_gen/runtime/layers/rotary_embedding/factory.py

@@ -5,8 +5,8 @@ from typing import Any
 
 import torch
 
-from ._base import LinearScalingRotaryEmbedding, RotaryEmbedding
+from .base import LinearScalingRotaryEmbedding, RotaryEmbedding
-from ._mrope import NDRotaryEmbedding, _to_tuple
+from .mrope import NDRotaryEmbedding, _to_tuple
 
 _ROPE_DICT: dict[tuple, RotaryEmbedding] = {}
 _ND_ROPE_CACHE: "OrderedDict[tuple, NDRotaryEmbedding]" = OrderedDict()