diff --git a/instruction-templates/Vigogne-Chat.yaml b/instruction-templates/Vigogne-Chat.yaml index 8fc3e45..4c4de1d 100644 --- a/instruction-templates/Vigogne-Chat.yaml +++ b/instruction-templates/Vigogne-Chat.yaml @@ -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 -%}