Skip to content

Commit 6c59a2c

Browse files
committed
support 1b model type
1 parent ad45bb3 commit 6c59a2c

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

llama.cpp

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -830,6 +830,7 @@ static llama_state g_state;
830830
// available llama models
831831
enum e_model {
832832
MODEL_UNKNOWN,
833+
MODEL_1B,
833834
MODEL_3B,
834835
MODEL_7B,
835836
MODEL_13B,
@@ -1626,6 +1627,7 @@ static void llm_load_hparams(
16261627
GGUF_GET_KEY(ctx, hparams.f_norm_eps, gguf_get_val_f32, GGUF_TYPE_FLOAT32, true, kv(LLM_KV_ATTENTION_LAYERNORM_EPS));
16271628

16281629
switch (hparams.n_layer) {
1630+
case 24: model.type = e_model::MODEL_1B; break;
16291631
case 32: model.type = e_model::MODEL_7B; break;
16301632
case 60: model.type = e_model::MODEL_40B; break;
16311633
default: model.type = e_model::MODEL_UNKNOWN;

0 commit comments

Comments
 (0)