Skip to content

Commit 40a9d39

Browse files
committed
Fix tensor name and reorder llm_types
1 parent 56adda7 commit 40a9d39

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

src/llama-model.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18071,7 +18071,7 @@ struct llm_build_seed_oss : public llm_graph_context {
1807118071
cur = build_norm(ffn_inp,
1807218072
model.layers[il].attn_post_norm, NULL,
1807318073
LLM_NORM_RMS, il);
18074-
cb(cur, "ffn_norm", il);
18074+
cb(cur, "attn_post_norm", il);
1807518075

1807618076
cur = build_ffn(cur,
1807718077
model.layers[il].ffn_up, NULL, NULL,

src/llama-model.h

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -76,6 +76,7 @@ enum llm_type {
7676
LLM_TYPE_32B,
7777
LLM_TYPE_34B,
7878
LLM_TYPE_35B,
79+
LLM_TYPE_36B, // Seed OSS
7980
LLM_TYPE_40B,
8081
LLM_TYPE_65B,
8182
LLM_TYPE_70B,
@@ -109,7 +110,6 @@ enum llm_type {
109110
LLM_TYPE_355B_A32B, // GLM-4.5
110111
LLM_TYPE_E2B,
111112
LLM_TYPE_E4B,
112-
LLM_TYPE_36B // Seed OSS
113113
};
114114

115115
std::string llama_rope_scaling_type_name(llama_rope_scaling_type rope_scaling_type);

0 commit comments

Comments
 (0)