rwkv integration completed

This commit is contained in:
Concedo
2023-05-28 00:48:56 +08:00
parent 55e0fbf024
commit 5d9f5b28a6
4 changed files with 99 additions and 35 deletions
+1 -1
View File
@@ -128,7 +128,7 @@ extern "C"
return true;
}
}
else if(file_format==FileFormat::RWKV_1)
else if(file_format==FileFormat::RWKV_1 || file_format==FileFormat::RWKV_2)
{
printf("\n---\nIdentified as RWKV model: (ver %d)\nAttempting to Load...\n---\n", file_format);
ModelLoadResult lr = gpttype_load_model(inputs, file_format);