Chat Template Fix

#4
by Hunterx - opened

Hi All - I just wanted to mention this post that fixes the thinking template for Kimi

https://huggingface.co/AesSedai/Kimi-K2.5-GGUF/discussions/1

"Just in case anybody else has problems with the thinking block not showing up, I posted the fix needed to the jinja template here:

https://huggingface.co/unsloth/Kimi-K2.5-GGUF/discussions/1#697b46fdf48287bb9c2e92dc

I've figured out what is causing this:

https://huggingface.co/moonshotai/Kimi-K2.5/blob/main/chat_template.jinja

{%- if add_generation_prompt -%}
<|im_assistant|>assistant<|im_middle|>
{%- if thinking is defined and thinking is false -%}

{%- else -%}
# <--- This forced tag doesn't set returned as part of the chat.
{%- endif -%}
{%- endif -%}

To fix this you need to make a copy of chat_template.jinja and delete the {%- else -%} part at the bottom like this:

{%- if add_generation_prompt -%}
<|im_assistant|>assistant<|im_middle|>
{%- if thinking is defined and thinking is false -%}

{%- endif -%}
{%- endif -%}

Then run llama.cpp using --jinja --chat-template-file chat_template.jinja options.

(Also: there doesn't seem to be any need/requirement to use --special option now)"

Here it is fixed for Open Claw if you need it for Open Claw:

command:
--jinja
--chat-template-file ./models/templates/moonshotai-Kimi-K25.jinja \ <---- save it as that in your models templates or anything really
--special

{%- macro render_content(msg) -%}
{%- set c = msg.get('content') -%}
{%- if c is string -%}
{{ c }}
{%- elif c is not none -%}
{% for content in c -%}
{% if content['type'] == 'image' or content['type'] == 'image_url' -%}
<|media_begin|>image<|media_content|><|media_pad|><|media_end|>
{% elif content['type'] == 'video' or content['type']== 'video_url'-%}
<|kimi_k25_video_placeholder|>
{% else -%}
{{ content['text'] }}
{%- endif -%}
{%- endfor -%}
{%- endif -%}
{%- endmacro -%}

{% macro set_roles(message) -%}
{%- set role_name = message.get('name') or message['role'] -%}
{%- if message['role'] == 'user' -%}
<|im_user|>{{role_name}}<|im_middle|>
{%- elif message['role'] == 'assistant' -%}
<|im_assistant|>{{role_name}}<|im_middle|>
{%- else -%}
<|im_system|>{{role_name}}<|im_middle|>
{%- endif -%}
{%- endmacro -%}

{%- macro render_toolcalls(message) -%}
<|tool_calls_section_begin|>
{%- for tool_call in message['tool_calls'] -%}
{%- set formatted_id = tool_call['id'] -%}
<|tool_call_begin|>{{ formatted_id }}<|tool_call_argument_begin|>
{%- if tool_call['function']['arguments'] is string -%}
{{ tool_call['function']['arguments'] }}
{%- else -%}
{{ tool_call['function']['arguments'] | tojson | replace(", ", ",") | replace(": ", ":") }}
{%- endif -%}
<|tool_call_end|>
{%- endfor -%}
<|tool_calls_section_end|>
{%- endmacro -%}

{# Find last non-tool-call assistant message #}
{%- set ns = namespace(last_non_tool_call_assistant_msg=-1) -%}
{%- for idx in range(messages|length-1, -1, -1) -%}
{%- if messages[idx]['role'] == 'assistant' and not messages[idx].get('tool_calls') -%}
{%- set ns.last_non_tool_call_assistant_msg = idx -%}
{%- break -%}
{%- endif -%}
{%- endfor -%}

{# Split messages #}
{%- set hist_msgs = messages[:ns.last_non_tool_call_assistant_msg+1] -%}
{%- set suffix_msgs = messages[ns.last_non_tool_call_assistant_msg+1:] -%}

{%- if tools -%}
{%- if tools_ts_str -%}
<|im_system|>tool_declare<|im_middle|>{{ tools_ts_str }}<|im_end|>
{%- else -%}
<|im_system|>tool_declare<|im_middle|>{{ tools | tojson }}<|im_end|>
{%- endif -%}
{%- endif -%}

{%- for message in hist_msgs -%}
{{set_roles(message)}}
{%- if message['role'] == 'assistant' -%}
{{render_content(message)}}
{%- if message.get('tool_calls') -%}
{{render_toolcalls(message)}}
{%- endif -%}
{%- elif message['role'] == 'tool' -%}
{%- set tool_call_id = message.tool_call_id -%}
## Return of {{ tool_call_id }}
{{render_content(message)}}
{%- elif message['content'] is not none -%}
{{render_content(message)}}
{%- endif -%}
<|im_end|>
{%- endfor -%}

{%- for message in suffix_msgs -%}
{{set_roles(message)}}
{%- if message['role'] == 'assistant' -%}
{%- if thinking is defined and thinking is false -%}
{{render_content(message)}}
{%- else -%}
{%- set rc = message.get('reasoning_content', '') -%}
{{rc}}{{render_content(message)}}
{%- endif -%}
{%- if message.get('tool_calls') -%}
{{render_toolcalls(message)}}
{%- endif -%}
{%- elif message['role'] == 'tool' -%}
{%- set tool_call_id = message.tool_call_id -%}
## Return of {{ tool_call_id }}
{{render_content(message)}}
{%- elif message['content'] is not none -%}
{{render_content(message)}}
{%- endif -%}
<|im_end|>
{%- endfor -%}

{%- if add_generation_prompt -%}
<|im_assistant|>assistant<|im_middle|>
{%- if thinking is defined and thinking is false -%}

{%- endif -%}
{%- endif -%}

Sign up or log in to comment