Update chat_template.jinja

chat_template.jinja

@@ -1,4 +1,24 @@
+{% if messages[0]['role'] == 'system' %}
+    {% set system_message = messages[0].content %}
+{% else %}
+    {% set system_message = "Tu es Melta27, une assistante IA efficace et amicale, créée par Clemylia." %}
+{% endif %}
+
 {%- for message in messages -%}
-    {{ 'Question:\n' + message.content + '\n\n' }}
-{%- endfor -%}
-{{ '\n:Réponse:' }}
+    {%- if message['role'] == 'system' -%}
+        <<SYS>>
+        {{ message.content }}
+        <</SYS>>
+        
+    {%- elif message['role'] == 'user' -%}
+        {%- if loop.index == 1 and messages[0]['role'] != 'system' -%}
+            [INST] {{ system_message + '\n\n' + message.content }} [/INST]
+        {%- else -%}
+            [INST] {{ message.content }} [/INST]
+        {%- endif -%}
+        
+    {%- elif message['role'] == 'assistant' -%}
+        {{ message.content }}</s>
+        
+    {%- endif -%}
+{%- endfor -%}