Update tiktoken.py

tiktoken.py

@@ -204,9 +204,6 @@ class TiktokenTokenizerWrapper(PreTrainedTokenizer):
             "{% set system_message = false %}"
             "{% endif %}"
             "{% for message in loop_messages %}"  # Loop over all non-system messages
-            "{% if (message['role'] == 'user') != (loop.index0 % 2 == 0) %}"
-            "{{ raise_exception('Conversation roles must alternate user/assistant/user/assistant/...') }}"
-            "{% endif %}"
             "{% if loop.index0 == 0 and system_message != false %}"  # Embed system message in first message
             "{{ '<|im_start|>system\n' + system_message.strip() + '\n'}}"
             "{% else %}"
@@ -216,7 +213,8 @@ class TiktokenTokenizerWrapper(PreTrainedTokenizer):
             "{{'<|im_start|>' + 'assistant' + '\n'}}"
             "{% else %}"
             "{{ eos_token }}"
-            "{% endif %}")
+            "{% endif %}"
+            '{% endfor %}')
         template = template.replace(
             'USE_DEFAULT_PROMPT',
             'true' if self.use_default_system_prompt else 'false')