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 @@ -5455,7 +5455,7 @@ static void llm_load_hparams(
5455
5455
// granite uses a vocab with len 49152
5456
5456
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;
5457
5457
case 36: model.type = e_model::MODEL_8B; break; // granite
5458
- case 40: model.type = hparams.n_vocab == 49152 ? e_model::MODEL_3B : e_model::MODEL_13B; break;
5458
+ case 40: model.type = ( hparams.n_vocab == 49152 || hparams.n_vocab == 49156) ? e_model::MODEL_3B : e_model::MODEL_13B; break;
5459
5459
case 48: model.type = e_model::MODEL_34B; break;
5460
5460
case 60: model.type = e_model::MODEL_30B; break;
5461
5461
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