force mistral think tokens to print

2026-08-31 09:30:50 +02:00 · 2026-02-11 21:26:16 +08:00
parent bb48a9a13b
commit f916966799
1 changed files with 7 additions and 0 deletions
@@ -2704,6 +2704,13 @@ void llama_vocab::impl::load(llama_model_loader & ml, const LLM_KV & kv) {

                attr = LLAMA_TOKEN_ATTR_USER_DEFINED;
            }
+
+            if(t.first=="[THINK]" || t.first=="[/THINK]" || t.first=="<think>" || t.first=="</think>")
+            {
+                 LLAMA_LOG_WARN("%s: setting token '%s' (%d) attribute to USER_DEFINED (%u), old attributes: %u\n",
+                        __func__, t.first.c_str(), t.second, LLAMA_TOKEN_ATTR_USER_DEFINED, attr);
+                attr = LLAMA_TOKEN_ATTR_USER_DEFINED;
+            }
        }

        // sanity checks