allow rwkv6 to run although its broken

2026-08-25 22:21:14 +02:00 · 2024-09-09 20:50:58 +08:00
parent b63158005f
commit fc7fe2e7a0
3 changed files with 25 additions and 8 deletions
@@ -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);

        }