From 306c479d3a8dd481df6734ef5e82b86ed300145b Mon Sep 17 00:00:00 2001
From: oobabooga <112222186+oobabooga@users.noreply.github.com>
Date: Sun, 17 Dec 2023 19:15:54 -0800
Subject: [PATCH] Minor fix to Vigogne-Chat template

---
 instruction-templates/Vigogne-Chat.yaml | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

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 -%}