zoeyzhi's picture
Prune LM head to last-token logits
895046f verified