Support Intern-S2-Preview (#24875)

2026-05-10 22:17:30 +08:00
parent 59faf986b2
commit 335dbd60b4
11 changed files with 64 additions and 2 deletions
@@ -76,6 +76,7 @@ def get_model_config(
        "Qwen3NextForCausalLM",
        "Qwen3VLMoeForConditionalGeneration",
        "Qwen3_5MoeForConditionalGeneration",
+        "InternS2PreviewForConditionalGeneration",
    ]:
        E = config.num_experts // ep_size
        topk = config.num_experts_per_tok