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 · Dec 18, 2023 · dc33178 · dc33178
1 parent d25c34c
commit dc33178
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|>"
     )
 )