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' -}} {{- '' + message['content'] + '\n' -}}
{%- else -%} {%- else -%}
{%- if message['role'] == 'user' -%} {%- if message['role'] == 'user' -%}
{{-'\n<|USER|>: ' + message['content'] + '\n'-}} {{-'<|USER|>: ' + message['content'] + '\n'-}}
{%- else -%} {%- else -%}
{{-'<|ASSISTANT|>: ' + message['content'] + '' -}} {{-'<|ASSISTANT|>: ' + message['content'] + '\n' -}}
{%- endif -%} {%- endif -%}
{%- endif -%} {%- endif -%}
{%- endfor -%} {%- endfor -%}