chat: Change role to user for user prompts (#1447)

Co-authored-by: Jack-Khuu <[email protected]>
pytorch · Jan 22, 2025 · fb65b8b · fb65b8b
1 parent 025d412
commit fb65b8b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torchchat/generate.py b/torchchat/generate.py
@@ -1126,7 +1126,7 @@ def chat(
                     messages_to_encode.append(
                         {"role": "system", "content": self.system_prompt}
                     )
-                messages_to_encode.append({"role": "system", "content": prompt})
+                messages_to_encode.append({"role": "user", "content": prompt})
                 encoded = self.chat_formatter.encode_dialog_prompt(
                     messages_to_encode, add_generation_prompt=True,
                 )