trl-internal-testing
/

tiny-DeepseekV4ForCausalLM

Text Generation

Model card Files Files and versions

Update chat_template.jinja

#6

by qgallouedec HF Staff - opened 9 days ago

base: refs/heads/main

←

from: refs/pr/6

Discussion Files changed

Files changed (1) hide show

chat_template.jinja +2 -2

chat_template.jinja CHANGED Viewed

@@ -142,7 +142,7 @@
     {%- elif msg.role == "assistant" -%}
         {%- set rc = msg.get("reasoning_content", "") or "" -%}
-        {%- if (thinking_mode == "thinking") and (not prev_has_task) and ((not effective_drop) or idx > fns.lu_idx) -%}
             {{- rc -}}{{- "</think>" -}}
         {%- endif -%}
         {{- msg.get("content", "") or "" -}}
@@ -190,7 +190,7 @@
     {%- elif need_transition and (msg.role == "user" or msg.role == "developer") and not (is_last and not add_generation_prompt) -%}
         {{- "<｜Assistant｜>" -}}
         {%- if thinking_mode == "thinking" -%}
-            {{- "<think>" if (not effective_drop) or idx >= fns.lu_idx else "</think>" -}}
         {%- else -%}
             {{- "</think>" -}}
         {%- endif -%}

     {%- elif msg.role == "assistant" -%}
         {%- set rc = msg.get("reasoning_content", "") or "" -%}
+        {%- if (thinking_mode == "thinking") and (not prev_has_task) -%}
             {{- rc -}}{{- "</think>" -}}
         {%- endif -%}
         {{- msg.get("content", "") or "" -}}
     {%- elif need_transition and (msg.role == "user" or msg.role == "developer") and not (is_last and not add_generation_prompt) -%}
         {{- "<｜Assistant｜>" -}}
         {%- if thinking_mode == "thinking" -%}
+            {{- "<think>" -}}
         {%- else -%}
             {{- "</think>" -}}
         {%- endif -%}