From 5a05c22162139b615d9ead29e2e4bba640c4c1f7 Mon Sep 17 00:00:00 2001 From: Chranos <826995883@qq.com> Date: Wed, 11 Feb 2026 11:40:57 +0800 Subject: [PATCH] add deepseekv3 --- .../vllm_mlu/model_executor/models/deepseek_v2.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/vllm-v0.6.2/vllm_mlu/vllm_mlu/model_executor/models/deepseek_v2.py b/vllm-v0.6.2/vllm_mlu/vllm_mlu/model_executor/models/deepseek_v2.py index f0b5c4e..d70cbfd 100644 --- a/vllm-v0.6.2/vllm_mlu/vllm_mlu/model_executor/models/deepseek_v2.py +++ b/vllm-v0.6.2/vllm_mlu/vllm_mlu/model_executor/models/deepseek_v2.py @@ -565,7 +565,9 @@ def vllm__module_executor__models__deepseek_v2__DeepseekV2ForCausalLM__load_weig @brief: add expert skiped condition and delete useless if name not in params_dict: continue condition ''' name = name.replace(weight_name, param_name) - if (("mlp.experts." in name or "mlp.shared_experts." in name or "mlp.shared_expert_gate." in name) + if (("mlp.experts." in name or "mlp.shared_experts." in name + or "mlp.shared_expert_gate." in name + or "e_score_correction_bias" in name) and name not in params_dict): continue ''' @@ -595,7 +597,9 @@ def vllm__module_executor__models__deepseek_v2__DeepseekV2ForCausalLM__load_weig if name.endswith(".bias") and name not in params_dict: continue - if (("mlp.experts." in name or "mlp.shared_experts." in name or "mlp.shared_expert_gate." in name) + if (("mlp.experts." in name or "mlp.shared_experts." in name + or "mlp.shared_expert_gate." in name + or "e_score_correction_bias" in name) and name not in params_dict): continue if is_pp_missing_parameter(name, self):