head: make models query-aware (last-token readout) + trainer update d9fbc8a Throstur commited on Feb 20
