train_direct: prepend user turn so Qwen chat template accepts it

2026-04-19 00:06:23 -04:00 · 2026-04-19 00:06:23 -04:00 · ce58a3507f
commit ce58a3507f
parent 8c59f46505
1 changed files with 4 additions and 1 deletions
--- 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,
        )