Public/koboldcpp
1
0
Fork 0
mirror of https://github.com/LostRuins/koboldcpp.git synced 2026-09-18 16:55:14 +02:00
Code Issues Packages Projects Releases Wiki Activity

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
model_adapter.cpp
+3 -2
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;
}