File tree 1 file changed +3
-0
lines changed 1 file changed +3
-0
lines changed Original file line number Diff line number Diff line change @@ -1811,6 +1811,7 @@ enum e_model {
1811
1811
MODEL_8x22B,
1812
1812
MODEL_16x12B,
1813
1813
MODEL_10B_128x3_66B,
1814
+ MODEL_57B_A14B,
1814
1815
};
1815
1816
1816
1817
static const size_t kiB = 1024;
@@ -3933,6 +3934,7 @@ static const char * llama_model_type_name(e_model type) {
3933
3934
case MODEL_8x22B: return "8x22B";
3934
3935
case MODEL_16x12B: return "16x12B";
3935
3936
case MODEL_10B_128x3_66B: return "10B+128x3.66B";
3937
+ case MODEL_57B_A14B: return "57B.A14B";
3936
3938
default: return "?B";
3937
3939
}
3938
3940
}
@@ -4250,6 +4252,7 @@ static void llm_load_hparams(
4250
4252
ml.get_key(LLM_KV_ATTENTION_LAYERNORM_RMS_EPS, hparams.f_norm_rms_eps);
4251
4253
switch (hparams.n_layer) {
4252
4254
case 24: model.type = e_model::MODEL_A2_7B; break;
4255
+ case 28: model.type = e_model::MODEL_57B_A14B; break;
4253
4256
default: model.type = e_model::MODEL_UNKNOWN;
4254
4257
}
4255
4258
} break;
You can’t perform that action at this time.
0 commit comments