We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent f34ffbc commit 94b62e7Copy full SHA for 94b62e7
src/llama-model.cpp
@@ -1271,7 +1271,12 @@ void llama_model::load_hparams(llama_model_loader & ml) {
1271
case 2048: type = LLM_TYPE_1_5B; break;
1272
default: type = LLM_TYPE_UNKNOWN;
1273
} break;
1274
- case 28: type = LLM_TYPE_7B; break; // ARWKV7
+ case 28:
1275
+ switch (hparams.n_embd) {
1276
+ case 1536: type = LLM_TYPE_1_5B; break;
1277
+ case 3584: type = LLM_TYPE_7B; break;
1278
+ default: type = LLM_TYPE_UNKNOWN;
1279
+ } break;
1280
case 32: type = LLM_TYPE_2_9B; break; // RWKV-7-World
1281
1282
}
0 commit comments