diff --git a/src/llama-model.cpp b/src/llama-model.cpp index a442abeb..a4f06112 100644 --- a/src/llama-model.cpp +++ b/src/llama-model.cpp @@ -6323,7 +6323,7 @@ struct llm_build_qwen2moe : public llm_graph_context { false, 0.0, LLAMA_EXPERT_GATING_FUNC_TYPE_SOFTMAX, il); - cb(cur, "ffn_moe_out", il); + cb(moe_out, "ffn_moe_out", il); // FFN shared expert {