From 56571c32385930d6abc18a52ed6fe795f41a3f70 Mon Sep 17 00:00:00 2001
From: alielfilali01 <alielfilali0909@gmail.com>
Date: Thu, 12 Feb 2026 08:22:12 +0000
Subject: [PATCH] fix: use NEOX RoPE type for JAIS2

---
 src/llama-model.cpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/llama-model.cpp b/src/llama-model.cpp
index 6adec5eebb..0eb6f71e94 100644
--- a/src/llama-model.cpp
+++ b/src/llama-model.cpp
@@ -8790,7 +8790,6 @@ llama_rope_type llama_model_rope_type(const llama_model * model) {
         case LLM_ARCH_ERNIE4_5:
         case LLM_ARCH_ERNIE4_5_MOE:
         case LLM_ARCH_MISTRAL3:
-        case LLM_ARCH_JAIS2:
         case LLM_ARCH_LLAMA_EMBED:
         case LLM_ARCH_MAINCODER:
             return LLAMA_ROPE_TYPE_NORM;
@@ -8841,6 +8840,7 @@ llama_rope_type llama_model_rope_type(const llama_model * model) {
         case LLM_ARCH_BAILINGMOE2:
         case LLM_ARCH_DOTS1:
         case LLM_ARCH_HUNYUAN_MOE:
+        case LLM_ARCH_JAIS2:
         case LLM_ARCH_OPENAI_MOE:
         case LLM_ARCH_HUNYUAN_DENSE:
         case LLM_ARCH_LFM2: