mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-01-11 23:00:51 +08:00
23 lines
631 B
Django/Jinja
23 lines
631 B
Django/Jinja
{% if messages[0]['role'] == 'system' %}
|
|
{% set offset = 1 %}
|
|
{% else %}
|
|
{% set offset = 0 %}
|
|
{% endif %}
|
|
|
|
{% for message in messages %}
|
|
{% if (message['role'] == 'user') != (loop.index0 % 2 == offset) %}
|
|
{{ raise_exception('Conversation roles must alternate user/bot/user/bot/...') }}
|
|
{% endif %}
|
|
|
|
{% if message['role'] == 'assistant' %}
|
|
{{ set role = 'bot' }}
|
|
{% else %}
|
|
{{ set role = message['role'] }}
|
|
{% endif %}
|
|
|
|
{{ bos_token + role + '\n' + message['content'] | trim + eos_token }}
|
|
{% endfor %}
|
|
|
|
{% if add_generation_prompt %}
|
|
{{ bos_token + 'bot\n' }}
|
|
{% endif %} |