Fix Bailing MoE model bugs (#10362)

Co-authored-by: luoyuan.luo <luoyuan.luo@antgroup.com> Co-authored-by: 羽癫 <yudian.zy@antgroup.com>
2025-09-12 15:36:02 +08:00
parent fac07c9b08
commit 24dc2bee97
2 changed files with 8 additions and 5 deletions
--- a/python/sglang/srt/server_args.py
+++ b/python/sglang/srt/server_args.py
@@ -757,7 +757,7 @@ class ServerArgs:
            if model_arch in [
                "DeepseekV3ForCausalLM",
                "Glm4MoeForCausalLM",
-                "BailingMoeV2ForCausalLM",
+                "BailingMoeForCausalLM",
                "BailingMoeV2ForCausalLM",
            ]:
                # Auto set draft_model_path DeepSeek-V3/R1