mirror of
https://github.com/vllm-project/vllm.git
synced 2026-06-06 00:16:14 +00:00
Dinghow Yang
18 lines
629 B
Django/Jinja
18 lines
629 B
Django/Jinja
{%- set counter = namespace(index=0) -%}
|
|
{%- for message in messages -%}
|
|
{%- if message['role'] == 'user' -%}
|
|
{{- '[Round ' + counter.index|string + ']\n问：' + message['content'] -}}
|
|
{%- set counter.index = counter.index + 1 -%}
|
|
{%- endif -%}
|
|
{%- if message['role'] == 'assistant' -%}
|
|
{{- '\n答：' + message['content'] -}}
|
|
{%- if (loop.last and add_generation_prompt) or not loop.last -%}
|
|
{{- '\n' -}}
|
|
{%- endif -%}
|
|
{%- endif -%}
|
|
{%- endfor -%}
|
|
|
|
|
|
{%- if add_generation_prompt and messages[-1]['role'] != 'assistant' -%}
|
|
{{- '\n答：' -}}
|
|
{%- endif -%} |