Skip to content

Commit e98b369

Browse files
authored
llama : set qwen3 model type sizes (#13175)
1 parent b6ce743 commit e98b369

File tree

2 files changed

+14
-0
lines changed

2 files changed

+14
-0
lines changed

src/llama-model.cpp

+10
Original file line numberDiff line numberDiff line change
@@ -92,6 +92,10 @@ const char * llm_type_name(llm_type type) {
9292
case LLM_TYPE_290B: return "290B";
9393
case LLM_TYPE_17B_16E: return "17Bx16E (Scout)";
9494
case LLM_TYPE_17B_128E: return "17Bx128E (Maverick)";
95+
case LLM_TYPE_0_6B: return "0.6B";
96+
case LLM_TYPE_1_7B: return "1.7B";
97+
case LLM_TYPE_30B_A3B: return "30B.A3B";
98+
case LLM_TYPE_235B_A22B: return "235B.A22B";
9599
default: return "?B";
96100
}
97101
}
@@ -793,6 +797,10 @@ void llama_model::load_hparams(llama_model_loader & ml) {
793797
{
794798
ml.get_key(LLM_KV_ATTENTION_LAYERNORM_RMS_EPS, hparams.f_norm_rms_eps);
795799
switch (hparams.n_layer) {
800+
case 28: type = hparams.n_embd == 1024 ? LLM_TYPE_0_6B : LLM_TYPE_1_7B; break;
801+
case 36: type = hparams.n_embd == 2560 ? LLM_TYPE_4B : LLM_TYPE_8B; break;
802+
case 40: type = LLM_TYPE_14B; break;
803+
case 64: type = LLM_TYPE_32B; break;
796804
default: type = LLM_TYPE_UNKNOWN;
797805
}
798806
} break;
@@ -802,6 +810,8 @@ void llama_model::load_hparams(llama_model_loader & ml) {
802810

803811
ml.get_key(LLM_KV_ATTENTION_LAYERNORM_RMS_EPS, hparams.f_norm_rms_eps);
804812
switch (hparams.n_layer) {
813+
case 48: type = LLM_TYPE_30B_A3B; break;
814+
case 94: type = LLM_TYPE_235B_A22B; break;
805815
default: type = LLM_TYPE_UNKNOWN;
806816
}
807817
} break;

src/llama-model.h

+4
Original file line numberDiff line numberDiff line change
@@ -88,6 +88,10 @@ enum llm_type {
8888
LLM_TYPE_290B,
8989
LLM_TYPE_17B_16E, // llama4 Scout
9090
LLM_TYPE_17B_128E, // llama4 Maverick
91+
LLM_TYPE_0_6B,
92+
LLM_TYPE_1_7B,
93+
LLM_TYPE_30B_A3B,
94+
LLM_TYPE_235B_A22B,
9195
};
9296

9397
struct llama_layer_posnet {

0 commit comments

Comments
 (0)