From 7857578f45f5358e6fac7d82564f225b7bd65e29 Mon Sep 17 00:00:00 2001
From: Concedo <39025047+LostRuins@users.noreply.github.com>
Date: Tue, 7 Oct 2025 13:47:15 +0800
Subject: [PATCH] handle more rnn models

---
 model_adapter.cpp | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/model_adapter.cpp b/model_adapter.cpp
index 81022bc5c..a3ca449d8 100644
--- a/model_adapter.cpp
+++ b/model_adapter.cpp
@@ -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;
             }