From 45b34dd507f0ca1182b132411b2571001de4e1a5 Mon Sep 17 00:00:00 2001 From: vlado Date: Mon, 30 Dec 2024 16:50:47 -0500 Subject: [PATCH] Change role to user for user prompts --- torchchat/generate.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/torchchat/generate.py b/torchchat/generate.py index e271f5027..11c8d2ad3 100644 --- a/torchchat/generate.py +++ b/torchchat/generate.py @@ -1122,7 +1122,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, )