"tests/vscode:/vscode.git/clone" did not exist on "292fc59d61ba97b9a79b9649e693d0cac2dea15f"
Unverified Commit b398e5c8 authored by Borushiki's avatar Borushiki Committed by GitHub
Browse files

Update get_expert_mapping to include self parameter (#33525)


Signed-off-by: default avatarBorushiki <38628261+Otsutsukii@users.noreply.github.com>
parent 78061ef5
......@@ -617,6 +617,7 @@ class Glm4MoeLiteForCausalLM(
# Params for weights, fp8 weight scales, fp8 activation scales
# (param_name, weight_name, expert_id, shard_id)
return SharedFusedMoE.make_expert_params_mapping(
self,
ckpt_gate_proj_name="gate_proj",
ckpt_down_proj_name="down_proj",
ckpt_up_proj_name="up_proj",
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment