pepijn 0e2dc1b76f fix(pi052): supervise only FAST action-code tokens ...

Mask the FAST auxiliary loss to discrete action-code tokens so wrapper formatting tokens do not affect action co-training.

Co-authored-by: Cursor <cursoragent@cursor.com>

2026-05-18 17:38:34 +00:00

..

test_pi052_attention_masking.py

fix(pi052): port the smolvla2 text-head fixes to pi052

2026-05-18 15:42:19 +02:00

test_pi052_fast_action_loss.py

fix(pi052): supervise only FAST action-code tokens

2026-05-18 17:38:34 +00:00

test_pi052_text_processor.py

fix(pi052): port the smolvla2 text-head fixes to pi052

2026-05-18 15:42:19 +02:00