File tree Expand file tree Collapse file tree 1 file changed +3
-0
lines changed Expand file tree Collapse file tree 1 file changed +3
-0
lines changed Original file line number Diff line number Diff line change @@ -2305,6 +2305,7 @@ enum e_model {
2305
2305
MODEL_1B,
2306
2306
MODEL_1_3B,
2307
2307
MODEL_1_4B,
2308
+ MODEL_1_5B,
2308
2309
MODEL_1_6B,
2309
2310
MODEL_2B,
2310
2311
MODEL_2_8B,
@@ -5231,6 +5232,7 @@ static const char * llama_model_type_name(e_model type) {
5231
5232
case MODEL_1B: return "1B";
5232
5233
case MODEL_1_3B: return "1.3B";
5233
5234
case MODEL_1_4B: return "1.4B";
5235
+ case MODEL_1_5B: return "1.5B";
5234
5236
case MODEL_1_6B: return "1.6B";
5235
5237
case MODEL_2B: return "2B";
5236
5238
case MODEL_2_8B: return "2.8B";
@@ -5602,6 +5604,7 @@ static void llm_load_hparams(
5602
5604
ml.get_key(LLM_KV_ATTENTION_LAYERNORM_RMS_EPS, hparams.f_norm_rms_eps);
5603
5605
switch (hparams.n_layer) {
5604
5606
case 24: model.type = hparams.n_embd == 1024 ? e_model::MODEL_0_5B : e_model::MODEL_1B; break;
5607
+ case 28: model.type = hparams.n_embd == 1536 ? e_model::MODEL_1_5B : e_model::MODEL_7B; break;
5605
5608
case 32: model.type = e_model::MODEL_7B; break;
5606
5609
case 40: model.type = hparams.n_head() == 20 ? e_model::MODEL_4B : e_model::MODEL_13B; break;
5607
5610
case 80: model.type = e_model::MODEL_70B; break;
You can’t perform that action at this time.
0 commit comments