vrr/koboldcpp
2
0
Fork 0
mirror of https://github.com/LostRuins/koboldcpp.git synced 2026-05-08 09:59:50 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

handle more rnn models

Browse source
This commit is contained in:
Concedo 2025-10-07 13:47:15 +08:00
parent bb5cef1756
commit 7857578f45
1 changed files with 3 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

5
model_adapter.cpp
View file

@ -367,7 +367,8 @@ std::string gguf_get_model_arch(const std::string & gguf_filename)
{
fileformatmeta->model_architecture = GGUFArch::ARCH_FALCON;
}
else if(modelarch=="mamba" || modelarch=="mamba2" || modelarch=="nemotron_h" || modelarch=="jamba") //lazy approach, put all RNN models
else if(modelarch=="mamba" || modelarch=="mamba2" || modelarch=="nemotron_h" || modelarch=="jamba" || modelarch=="granitehybrid" || modelarch=="lfm2"
|| modelarch=="plamo2" || modelarch=="falcon-h1") //lazy approach, put all RNN models
{
fileformatmeta->model_architecture = GGUFArch::ARCH_MAMBALIKE;
}
@ -391,7 +392,7 @@ std::string gguf_get_model_arch(const std::string & gguf_filename)
{
fileformatmeta->model_architecture = GGUFArch::ARCH_GEMMA3N;
}
else if(modelarch=="rwkv6" || modelarch=="rwkv7")
else if(modelarch=="rwkv6" || modelarch=="rwkv7" || modelarch=="rwkv6qwen2" || modelarch=="arwkv7")
{
fileformatmeta->model_architecture = GGUFArch::ARCH_RWKV;
}