From 06d0a3d92b25ee24e33cd8150d738b0f5a870889 Mon Sep 17 00:00:00 2001 From: Xuchun Shang Date: Thu, 17 Apr 2025 17:41:41 +0800 Subject: [PATCH] [Bug fix] use correct func path in deepseek (#5496) Signed-off-by: Xuchun Shang --- python/sglang/srt/models/deepseek.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/sglang/srt/models/deepseek.py b/python/sglang/srt/models/deepseek.py index 2b963be16..1664f17ff 100644 --- a/python/sglang/srt/models/deepseek.py +++ b/python/sglang/srt/models/deepseek.py @@ -170,7 +170,7 @@ class DeepseekMoE(nn.Module): shared_output = self.shared_experts(hidden_states) # router_logits: (num_tokens, n_experts) router_logits, _ = self.gate(hidden_states) - final_hidden_states = fused_moe( + final_hidden_states = fused_moe.fused_moe( hidden_states, self.w1, self.w2,