Skip to content

Commit 7857578

Browse files
committed
handle more rnn models
1 parent bb5cef1 commit 7857578

1 file changed

Lines changed: 3 additions & 2 deletions

File tree

model_adapter.cpp

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -367,7 +367,8 @@ std::string gguf_get_model_arch(const std::string & gguf_filename)
367367
{
368368
fileformatmeta->model_architecture = GGUFArch::ARCH_FALCON;
369369
}
370-
else if(modelarch=="mamba" || modelarch=="mamba2" || modelarch=="nemotron_h" || modelarch=="jamba") //lazy approach, put all RNN models
370+
else if(modelarch=="mamba" || modelarch=="mamba2" || modelarch=="nemotron_h" || modelarch=="jamba" || modelarch=="granitehybrid" || modelarch=="lfm2"
371+
|| modelarch=="plamo2" || modelarch=="falcon-h1") //lazy approach, put all RNN models
371372
{
372373
fileformatmeta->model_architecture = GGUFArch::ARCH_MAMBALIKE;
373374
}
@@ -391,7 +392,7 @@ std::string gguf_get_model_arch(const std::string & gguf_filename)
391392
{
392393
fileformatmeta->model_architecture = GGUFArch::ARCH_GEMMA3N;
393394
}
394-
else if(modelarch=="rwkv6" || modelarch=="rwkv7")
395+
else if(modelarch=="rwkv6" || modelarch=="rwkv7" || modelarch=="rwkv6qwen2" || modelarch=="arwkv7")
395396
{
396397
fileformatmeta->model_architecture = GGUFArch::ARCH_RWKV;
397398
}

0 commit comments

Comments
 (0)