Spaces:

tjtanaa
/

chat-template-generation

Runtime error

App Files Files Community

chat-template-generation / templates /airoboros_v2.jinja2

tjtanaa's picture

init streamlit chat-template ui

7a664fd 7 months ago

raw history blame contribute delete

No virus

491 Bytes

	{% if not add_generation_prompt is defined %}
	{% set add_generation_prompt = false %}
	{% endif %}
	{% for message in messages %}
	{% if not loop.first %}
	{% endif %}
	{% if message['role'] == 'system' %}
	{{ message['content'] + '\n' }}
	{% elif message['role'] == 'user' %}
	{{ 'USER: ' + message['content'] + '\n' }}
	{% elif message['role'] == 'assistant' %}
	{{ 'ASSISTANT: ' + message['content'] + '</s>' }}
	{% endif %}
	{% endfor %}
	{% if add_generation_prompt %}
	{{ 'ASSISTANT:' }}
	{% endif %}