Update chat template

tokenizer_config.json

@@ -169,7 +169,7 @@
     ]
   },
   "bos_token": "<s>",
-  "chat_template": "{{ bos_token }}{% for message in messages %}{{'<|im_start|>' + message['role'] + '\n' + message['content'] + '<|im_end|>' + '\n'}}{% endfor %}{% if add_generation_prompt %}{{ '<|im_start|>assistant\n' }}{% endif %}",
+  "chat_template": "{{ bos_token }}{% for message in messages %}\n    {%- if message['role'] == 'user' and template -%}\n        {{- '<|im_start|>' + message['role'] -}}\n        {{ '\n# Template:' }}\n        {{- '\n' + template }}\n        {% if examples %}\n            {{- '# Examples:' }}\n            {% for example in examples %}\n                {{- '## Input:\n' }}\n                {{- example['input'] + '\n' }}\n                {{- '## Output:\n' }}\n                {{- example['output'] | trim }}\n            {% endfor %}\n        {%- endif %}\n        {{- '# Context:' }}\n        {% if message['content'] is string %}\n            {{- message['content'] | trim }}\n        {% else %}\n            {% for content in message['content'] %}\n                {%- if content is string %}\n                    {{- content | trim }}\n                {%- elif content['type'] == 'text' %}\n                    {{- content['text'] | trim }}\n                {%- endif %}\n            {% endfor %}\n        {% endif %}\n        {{- '<|im_end|> '}}\n    {% else %}\n        {{- '<|im_start|>' + message['role'] }}\n        {% if message['content'] is string %}\n            {{- message['content'] | trim }}\n        {% else %}\n            {% for content in message['content'] %}\n                {%- if content is string %}\n                    {{- content | trim }}\n                {%- elif content['type'] == 'text' %}\n                    {{- content['text'] | trim }}\n                {%- endif %}\n            {% endfor %}\n        {% endif %}\n        {{- '<|im_end|> '}}\n    {% endif %}\n{% endfor -%}\n{%- if add_generation_prompt %}\n    {{- '<|im_start|>assistant' }}\n{% endif -%}\n",
   "clean_up_tokenization_spaces": false,
   "eos_token": "</s>",
   "extra_special_tokens": {},
@@ -177,4 +177,4 @@
   "pad_token": "</s>",
   "tokenizer_class": "InternLM2Tokenizer",
   "unk_token": "<unk>"
-}
+}