Minor fix to Vigogne-Chat template

This commit is contained in:
oobabooga 2023-12-17 19:15:54 -08:00
parent 3f973e1fbf
commit 306c479d3a

View File

@ -13,9 +13,9 @@ instruction_template: |-
{{- '' + message['content'] + '\n' -}}
{%- else -%}
{%- if message['role'] == 'user' -%}
{{-'\n<|USER|>: ' + message['content'] + '\n'-}}
{{-'<|USER|>: ' + message['content'] + '\n'-}}
{%- else -%}
{{-'<|ASSISTANT|>: ' + message['content'] + '' -}}
{{-'<|ASSISTANT|>: ' + message['content'] + '\n' -}}
{%- endif -%}
{%- endif -%}
{%- endfor -%}