model : add gpt-oss type strings (#15424)
This commit is contained in:
parent
1e19f5d462
commit
9ef6b0b835
|
|
@ -86,6 +86,7 @@ const char * llm_type_name(llm_type type) {
|
||||||
case LLM_TYPE_40B: return "40B";
|
case LLM_TYPE_40B: return "40B";
|
||||||
case LLM_TYPE_65B: return "65B";
|
case LLM_TYPE_65B: return "65B";
|
||||||
case LLM_TYPE_70B: return "70B";
|
case LLM_TYPE_70B: return "70B";
|
||||||
|
case LLM_TYPE_120B: return "120B";
|
||||||
case LLM_TYPE_142B: return "142B";
|
case LLM_TYPE_142B: return "142B";
|
||||||
case LLM_TYPE_236B: return "236B";
|
case LLM_TYPE_236B: return "236B";
|
||||||
case LLM_TYPE_290B: return "290B";
|
case LLM_TYPE_290B: return "290B";
|
||||||
|
|
@ -1834,7 +1835,11 @@ void llama_model::load_hparams(llama_model_loader & ml) {
|
||||||
hparams.swa_type = LLAMA_SWA_TYPE_STANDARD;
|
hparams.swa_type = LLAMA_SWA_TYPE_STANDARD;
|
||||||
hparams.set_swa_pattern(2);
|
hparams.set_swa_pattern(2);
|
||||||
|
|
||||||
// TODO: switch (hparams.n_layer)
|
switch (hparams.n_layer) {
|
||||||
|
case 24: type = LLM_TYPE_20B; break;
|
||||||
|
case 36: type = LLM_TYPE_120B; break;
|
||||||
|
default: type = LLM_TYPE_UNKNOWN;
|
||||||
|
}
|
||||||
} break;
|
} break;
|
||||||
case LLM_ARCH_LFM2:
|
case LLM_ARCH_LFM2:
|
||||||
{
|
{
|
||||||
|
|
|
||||||
|
|
@ -79,6 +79,7 @@ enum llm_type {
|
||||||
LLM_TYPE_40B,
|
LLM_TYPE_40B,
|
||||||
LLM_TYPE_65B,
|
LLM_TYPE_65B,
|
||||||
LLM_TYPE_70B,
|
LLM_TYPE_70B,
|
||||||
|
LLM_TYPE_120B,
|
||||||
LLM_TYPE_142B,
|
LLM_TYPE_142B,
|
||||||
LLM_TYPE_236B,
|
LLM_TYPE_236B,
|
||||||
LLM_TYPE_290B,
|
LLM_TYPE_290B,
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue