From ca9abff3e128217242339868704f0479e71f3bad Mon Sep 17 00:00:00 2001
From: Wing Lian <wing.lian@gmail.com>
Date: Mon, 25 Sep 2023 14:18:08 -0400
Subject: [PATCH] ensure we correctly set system message

---
 src/axolotl/prompters.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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