biondizzle/vllm
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

Update get_expert_mapping to include self parameter (#33525)

Browse Source 
Signed-off-by: Borushiki <38628261+Otsutsukii@users.noreply.github.com>
This commit is contained in:
Borushiki
2026-02-02 13:29:07 +01:00
committed by GitHub
parent 78061ef584
commit b398e5c819
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
vllm/model_executor/models/glm4_moe_lite.py
View File

@@ -617,6 +617,7 @@ class Glm4MoeLiteForCausalLM(
# Params for weights, fp8 weight scales, fp8 activation scales # Params for weights, fp8 weight scales, fp8 activation scales
# (param_name, weight_name, expert_id, shard_id) # (param_name, weight_name, expert_id, shard_id)
return SharedFusedMoE.make_expert_params_mapping( return SharedFusedMoE.make_expert_params_mapping(
self,
ckpt_gate_proj_name="gate_proj", ckpt_gate_proj_name="gate_proj",
ckpt_down_proj_name="down_proj", ckpt_down_proj_name="down_proj",
ckpt_up_proj_name="up_proj", ckpt_up_proj_name="up_proj",