ensure we correctly set system message

axolotl-ai-cloud · Sep 25, 2023 · ca9abff · ca9abff
1 parent 7763abe
commit ca9abff
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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]}