allow rwkv6 to run although its broken

This commit is contained in:
Concedo 2024-09-09 20:50:58 +08:00
parent b63158005f
commit fc7fe2e7a0
3 changed files with 25 additions and 8 deletions

View file

@ -314,6 +314,10 @@ void print_tok_vec(std::vector<float> &embd)
{
fileformatmeta->model_architecture = GGUFArch::ARCH_QWEN2;
}
else if(modelarch=="rwkv6")
{
fileformatmeta->model_architecture = GGUFArch::ARCH_RWKV;
}
printf("Arch Category: %d\n",fileformatmeta->model_architecture);
}