File tree Expand file tree Collapse file tree 2 files changed +7
-4
lines changed Expand file tree Collapse file tree 2 files changed +7
-4
lines changed Original file line number Diff line number Diff line change @@ -154,6 +154,7 @@ const char * llm_type_name(llm_type type) {
154154 case LLM_TYPE_1_7B: return " 1.7B" ;
155155 case LLM_TYPE_1_8B: return " 1.8B" ;
156156 case LLM_TYPE_2B: return " 2B" ;
157+ case LLM_TYPE_2_6B: return " 2.6B" ;
157158 case LLM_TYPE_2_8B: return " 2.8B" ;
158159 case LLM_TYPE_2_9B: return " 2.9B" ;
159160 case LLM_TYPE_3B: return " 3B" ;
@@ -2089,10 +2090,11 @@ void llama_model::load_hparams(llama_model_loader & ml) {
20892090 for (uint32_t il = 0 ; il < hparams.n_layer ; ++il) {
20902091 hparams.recurrent_layer_arr [il] = hparams.n_head_kv (il) == 0 ;
20912092 }
2092- switch (hparams.n_embd ) {
2093- case 1024 : type = LLM_TYPE_350M; break ;
2094- case 1536 : type = LLM_TYPE_700M; break ;
2095- case 2048 : type = LLM_TYPE_1_2B; break ;
2093+ switch (hparams.n_ff ()) {
2094+ case 4608 : type = LLM_TYPE_350M; break ;
2095+ case 6912 : type = LLM_TYPE_700M; break ;
2096+ case 8192 : type = LLM_TYPE_1_2B; break ;
2097+ case 10752 : type = LLM_TYPE_2_6B; break ;
20962098 default : type = LLM_TYPE_UNKNOWN;
20972099 }
20982100 } break ;
Original file line number Diff line number Diff line change @@ -58,6 +58,7 @@ enum llm_type {
5858 LLM_TYPE_1_7B,
5959 LLM_TYPE_1_8B,
6060 LLM_TYPE_2B,
61+ LLM_TYPE_2_6B,
6162 LLM_TYPE_2_8B,
6263 LLM_TYPE_2_9B,
6364 LLM_TYPE_3B,
You can’t perform that action at this time.
0 commit comments