add audio tags

chat_template.jinja

@@ -1,9 +1,14 @@
 {% for m in messages %}
-{% if m['role'] == 'system' %}<|im_start|>system
-{{ m['content'] }}<|im_end|>
-{% elif m['role'] == 'user' %}<|im_start|>user
-{{ m['content'] | replace('<audio>', '<|audio_start|>') | replace('</audio>', '<|audio_end|>') }}<|im_end|>
-{% else %}<|im_start|>assistant
-{{ m['content'] }}<|im_end|>
+<|im_start|>{{ m['role'] }}
+{% for piece in m['content'] %}
+{% if piece['type'] == 'text' %}
+{{ piece['text'] }}
+{% elif piece['type'] == 'audio' %}
+<|audio_start|><|audio_pad|><|audio_end|>
+{% elif piece['type'] == 'image' %}
+<image>
+{% elif piece['type'] == 'video' %}
+<video>
 {% endif %}
-{% endfor %}
+{% endfor %}<|im_end|>
+{% endfor %}