Skip to content

Commit 0458ff8

Browse files
fix bug
1 parent 9f8a776 commit 0458ff8

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/llama-arch.cpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1168,8 +1168,8 @@ static const std::map<llm_arch, std::map<llm_tensor, const char *>> LLM_TENSOR_N
11681168
{ LLM_TENSOR_FFN_NORM, "blk.%d.ffn_norm" },
11691169
{ LLM_TENSOR_FFN_UP, "blk.%d.ffn_up" },
11701170
{ LLM_TENSOR_FFN_DOWN, "blk.%d.ffn_down" },
1171-
{ LLM_TENSOR_ATTN_POST_NORM, "blk.%d.post_attn_norm" },
1172-
{ LLM_TENSOR_FFN_POST_NORM, "blk.%d.post_mlp_norm" },
1171+
{ LLM_TENSOR_ATTN_POST_NORM, "blk.%d.post_attention_norm" },
1172+
{ LLM_TENSOR_FFN_POST_NORM, "blk.%d.post_ffw_norm" },
11731173
},
11741174
},
11751175
{

0 commit comments

Comments
 (0)