model : add gpt-oss type strings (#15424)

This commit is contained in:
Georgi Gerganov 2025-08-19 19:58:28 +03:00 committed by GitHub
parent 1e19f5d462
commit 9ef6b0b835
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 7 additions and 1 deletions

View File

@ -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:
{ {

View File

@ -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,