Update README.md

README.md

@@ -22,9 +22,11 @@ tokenizer.chat_template = """{% if messages[0]['role'] == 'system' %}
 {% set loop_messages = messages %}
 {% set system_message = false %}
 {% endif %}
+{% if not (messages[0]['role'] == 'assistant' and loop_messages|length > 0) %}
 {{ bos_token }}
+{% endif %}
 {% for message in loop_messages %}
-{% if (message['role'] == 'user') != ((loop.index0 + messages[0]['role'] == 'assistant') % 2 == 0) %}
+{% if (message['role'] == 'user') != ((loop.index0 + (1 if messages[0]['role'] == 'assistant' else 0)) % 2 == 0) %}
 {{ raise_exception('Conversation roles must alternate starting from the first role.') }}
 {% endif %}
 {% if loop.index0 == 0 and system_message != false %}
@@ -39,8 +41,7 @@ tokenizer.chat_template = """{% if messages[0]['role'] == 'system' %}
 {% elif message['role'] == 'assistant' %}
 {{ ''  + content.strip() + '' + eos_token }}
 {% endif %}
-{% endfor %}
-"""
+{% endfor %}"""
 ```
 元のモデルのrevisionは`8b17f1c87697fb354952fa0d1018568e50bdff56`です。