From ce58a3507f044bb3e5930dc3bd713f6709dd64f0 Mon Sep 17 00:00:00 2001 From: ProofOfConcept Date: Sun, 19 Apr 2026 00:06:23 -0400 Subject: [PATCH] train_direct: prepend user turn so Qwen chat template accepts it --- training/amygdala_training/train_direct.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/training/amygdala_training/train_direct.py b/training/amygdala_training/train_direct.py index 02792b3..8749e37 100644 --- a/training/amygdala_training/train_direct.py +++ b/training/amygdala_training/train_direct.py @@ -86,7 +86,10 @@ def main() -> None: def apply_template(text: str) -> str: return tokenizer.apply_chat_template( - [{"role": "assistant", "content": text}], + [ + {"role": "user", "content": "How do you feel right now?"}, + {"role": "assistant", "content": text}, + ], tokenize=False, )