Skip to content

Commit 9d5d882

Browse files
authored
model : Add label for LFM2.5-230M (ggml-org#25008)
1 parent 1ec44d1 commit 9d5d882

3 files changed

Lines changed: 3 additions & 0 deletions

File tree

src/llama-model.cpp

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -700,6 +700,7 @@ const char * llm_type_name(llm_type type) {
700700
case LLM_TYPE_160M: return "160M";
701701
case LLM_TYPE_190M: return "190M";
702702
case LLM_TYPE_220M: return "220M";
703+
case LLM_TYPE_230M: return "230M";
703704
case LLM_TYPE_250M: return "250M";
704705
case LLM_TYPE_256M: return "256M";
705706
case LLM_TYPE_270M: return "270M";

src/llama-model.h

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -36,6 +36,7 @@ enum llm_type {
3636
LLM_TYPE_160M,
3737
LLM_TYPE_190M,
3838
LLM_TYPE_220M,
39+
LLM_TYPE_230M,
3940
LLM_TYPE_250M,
4041
LLM_TYPE_256M,
4142
LLM_TYPE_270M,

src/models/lfm2.cpp

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -13,6 +13,7 @@ void llama_model_lfm2::load_arch_hparams(llama_model_loader & ml) {
1313
hparams.n_layer_dense_lead = hparams.n_layer();
1414

1515
switch (hparams.n_ff()) {
16+
case 2560: type = LLM_TYPE_230M; break;
1617
case 4608: type = LLM_TYPE_350M; break;
1718
case 6912: type = LLM_TYPE_700M; break;
1819
case 8192: type = LLM_TYPE_1_2B; break;

0 commit comments

Comments
 (0)