diff --git a/src/axolotl/prompters.py b/src/axolotl/prompters.py index 5b955cc0b9..3310f08389 100644 --- a/src/axolotl/prompters.py +++ b/src/axolotl/prompters.py @@ -250,7 +250,7 @@ def build_prompt(self, source) -> Generator[str, None, None]: # Add the conversation system prompt if provided, otherwise use the default one if source[0]["from"] == "system": - conv.system = source[0]["value"] + conv.set_system_message(source[0]["value"]) source.pop(0) roles = {"human": conv.roles[0], "gpt": conv.roles[1]}