Update sharegpt.py

Resolves the double EOS token issue at the end of prompts when using Chatml template with shareGPT.py. axolotl-ai-cloud#922 (comment)
noobmaster29 · Jan 6, 2024 · d4eb95f · d4eb95f
1 parent 553c80f
commit d4eb95f
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
@@ -12,8 +12,8 @@
         system_template="<|im_start|>system\n{system_message}",
         system_message="You are a helpful assistant.",
         roles=["<|im_start|>user", "<|im_start|>assistant"],
-        sep_style=SeparatorStyle.CHATML,
         sep="<|im_end|>",
+        stop_str="<|im_end|>"
     )
 )