Update MFTCoder chat template

chencyudel · web-flow · commit 46e772f97928 · 2024-02-18T13:20:44.000+08:00
diff --git a/mftcoder_accelerate/src/tokenizer/chat_template.py b/mftcoder_accelerate/src/tokenizer/chat_template.py
@@ -42,15 +42,15 @@
     "{% set content = '' %}"
     "{% endif %}"
     "{% if message['role'] == 'user' or message['role'] == 'human' %}"
-    "{{ content + '<s>user\n' + message['content'] + '\n' }}"
+    "{{ content + '<s>human\n' + message['content'] + '\n' }}"
     "{% elif message['role'] == 'assistant' or message['role'] == 'bot' %}"
-    "{{ '<s>assistant\n' + message['content'] + '\n' +  eos_token + '\n'}}"
+    "{{ '<s>bot\n' + message['content'] + '\n' +  eos_token + '\n'}}"
     "{% else %}"
     "{{ raise_exception('Only user/human and assistant/bot roles are supported!') }}"
     "{% endif %}"
     "{% endfor %}"
     "{% if add_generation_prompt %}"
-    "{{ '<s>assistant\n' }}"
+    "{{ '<s>bot\n' }}"
     "{% endif %}"
 )
 
