Remove train_state from historical pi05 checkpoints b0b0edc verified HanchuZhou commited on 7 days ago