kent/consciousness
1
0
Fork 1
Code Issues 1 Pull requests 5 Projects Releases Packages Wiki Activity Actions

train_direct: prepend user turn so Qwen chat template accepts it

Browse source
This commit is contained in:
ProofOfConcept 2026-04-19 00:06:23 -04:00
parent 8c59f46505
commit ce58a3507f
1 changed files with 4 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

5
training/amygdala_training/train_direct.py
View file

@ -86,7 +86,10 @@ def main() -> None:
def apply_template(text: str) -> str: def apply_template(text: str) -> str:
return tokenizer.apply_chat_template( return tokenizer.apply_chat_template(
[{"role": "assistant", "content": text}], [
{"role": "user", "content": "How do you feel right now?"},
{"role": "assistant", "content": text},
],
tokenize=False, tokenize=False,
) )