File tree Expand file tree Collapse file tree 1 file changed +1
-1
lines changed Expand file tree Collapse file tree 1 file changed +1
-1
lines changed Original file line number Diff line number Diff line change @@ -5401,7 +5401,7 @@ static void llm_load_hparams(
5401
5401
// granite uses a vocab with len 49152
5402
5402
case 32: model.type = hparams.n_vocab == 49152 ? e_model::MODEL_3B : (hparams.n_vocab < 40000 ? e_model::MODEL_7B : e_model::MODEL_8B); break;
5403
5403
case 36: model.type = e_model::MODEL_8B; break; // granite
5404
- case 40: model.type = hparams.n_vocab == 49152 ? e_model::MODEL_3B : e_model::MODEL_13B; break;
5404
+ case 40: model.type = ( hparams.n_vocab == 49152 || hparams.n_vocab == 49156) ? e_model::MODEL_3B : e_model::MODEL_13B; break;
5405
5405
case 48: model.type = e_model::MODEL_34B; break;
5406
5406
case 60: model.type = e_model::MODEL_30B; break;
5407
5407
case 80: model.type = hparams.n_head() == hparams.n_head_kv() ? e_model::MODEL_65B : e_model::MODEL_70B; break;
You can’t perform that action at this time.
0 commit comments