Skip to content

Commit 94b62e7

Browse files
committed
llama: add more (a)rwkv7 variants in size
Signed-off-by: Molly Sophia <[email protected]>
1 parent f34ffbc commit 94b62e7

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

src/llama-model.cpp

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1271,7 +1271,12 @@ void llama_model::load_hparams(llama_model_loader & ml) {
12711271
case 2048: type = LLM_TYPE_1_5B; break;
12721272
default: type = LLM_TYPE_UNKNOWN;
12731273
} break;
1274-
case 28: type = LLM_TYPE_7B; break; // ARWKV7
1274+
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;
12751280
case 32: type = LLM_TYPE_2_9B; break; // RWKV-7-World
12761281
default: type = LLM_TYPE_UNKNOWN;
12771282
}

0 commit comments

Comments
 (0)