add a newline after im_end token

axolotl-ai-cloud · Sep 25, 2023 · 7763abe · 7763abe
1 parent b2d5daa
commit 7763abe
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/axolotl/prompt_strategies/sharegpt.py b/src/axolotl/prompt_strategies/sharegpt.py
@@ -13,7 +13,7 @@
         system_message="You are a helpful assistant.",
         roles=["<|im_start|>user", "<|im_start|>assistant"],
         sep_style=SeparatorStyle.CHATML,
-        sep="<|im_end|>",
+        sep="<|im_end|>\n",
     )
 )