Skip to content

Commit 261775d

Browse files
committed
add "glm4_moe" LLM_ARCH
1 parent d85099e commit 261775d

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

src/llama-arch.cpp

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -62,6 +62,7 @@ static const std::map<llm_arch, const char *> LLM_ARCH_NAMES = {
6262
{ LLM_ARCH_DEEPSEEK2, "deepseek2" },
6363
{ LLM_ARCH_CHATGLM, "chatglm" },
6464
{ LLM_ARCH_GLM4, "glm4" },
65+
{ LLM_ARCH_GLM4_MOE, "glm4_moe" },
6566
{ LLM_ARCH_BITNET, "bitnet" },
6667
{ LLM_ARCH_T5, "t5" },
6768
{ LLM_ARCH_T5ENCODER, "t5encoder" },
@@ -1398,8 +1399,8 @@ static const std::map<llm_arch, std::map<llm_tensor, const char *>> LLM_TENSOR_N
13981399
{ LLM_TENSOR_OUTPUT_NORM, "output_norm" },
13991400
{ LLM_TENSOR_OUTPUT, "output" },
14001401
{ LLM_TENSOR_ATTN_NORM, "blk.%d.attn_norm" },
1401-
{ LLM_TENSOR_ATTN_Q_NORM, "blk.%d.attn_q_norm" },
1402-
{ LLM_TENSOR_ATTN_K_NORM, "blk.%d.attn_k_norm" },
1402+
{ LLM_TENSOR_ATTN_Q_NORM, "blk.%d.attn_q_norm" }, // optional
1403+
{ LLM_TENSOR_ATTN_K_NORM, "blk.%d.attn_k_norm" }, // optional
14031404
{ LLM_TENSOR_ATTN_Q, "blk.%d.attn_q" },
14041405
{ LLM_TENSOR_ATTN_K, "blk.%d.attn_k" },
14051406
{ LLM_TENSOR_ATTN_V, "blk.%d.attn_v" },

0 commit comments

Comments
 (0)