17 lines
558 B
Django/Jinja
17 lines
558 B
Django/Jinja
{%- for message in messages -%}
|
|
{%- if message['role'] == 'system' -%}
|
|
{{- 'System: ' + message['content'] -}}
|
|
{%- elif message['role'] == 'user' -%}
|
|
{{- 'User: ' + message['content'] -}}
|
|
{%- elif message['role'] == 'assistant' -%}
|
|
{{- 'Falcon: ' + message['content'] -}}
|
|
{%- endif -%}
|
|
{%- if (loop.last and add_generation_prompt) or not loop.last -%}
|
|
{{- '\n' -}}
|
|
{%- endif -%}
|
|
{%- endfor -%}
|
|
|
|
|
|
{%- if add_generation_prompt and messages[-1]['role'] != 'assistant' -%}
|
|
{{- 'Falcon:' -}}
|
|
{% endif %} |