Skip to content

Commit 3714c3e

Browse files
authored
llama : fix incorrect Qwen2Moe ffn_moe_out graph callback (ggml-org#12631)
1 parent b4ae508 commit 3714c3e

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/llama-model.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6323,7 +6323,7 @@ struct llm_build_qwen2moe : public llm_graph_context {
63236323
false, 0.0,
63246324
LLAMA_EXPERT_GATING_FUNC_TYPE_SOFTMAX,
63256325
il);
6326-
cb(cur, "ffn_moe_out", il);
6326+
cb(moe_out, "ffn_moe_out", il);
63276327

63286328
// FFN shared expert
63296329
{

0 commit comments

Comments
 (0)