Reorder PD disagg CI tests (#11438)

2025-10-10 17:56:49 -07:00
parent 92777135a0
commit 61055cb309
7 changed files with 402 additions and 152 deletions
--- a/python/sglang/srt/utils/common.py
+++ b/python/sglang/srt/utils/common.py
@@ -523,7 +523,7 @@ def make_layers_non_pp(
    layer_fn: LayerFn,
    prefix: str = "",
 ) -> torch.nn.ModuleList:
-    from sglang.srt.offloader import get_offloader
+    from sglang.srt.utils.offloader import get_offloader

    layers = torch.nn.ModuleList(
        get_offloader().wrap_modules(