Fix swa memory pool size with spec (#17630)

2026-01-25 14:10:43 +08:00
parent 1674b9ef44
commit 30ece5e1d6
3 changed files with 48 additions and 49 deletions
--- a/python/sglang/srt/configs/model_config.py
+++ b/python/sglang/srt/configs/model_config.py
@@ -1259,7 +1259,8 @@ def get_hybrid_layer_ids(
            i for i in range(num_hidden_layers) if hybrid_layer_pattern[i] == 0
        ]
    elif "MiMoV2MTP" in model_architectures:
-        return [0], []
+        swa_attention_layer_ids = [0]
+        full_attention_layer_ids = []
    else:
        swa_attention_layer_ids = None
        full_attention_layer_ids = None