Generate tokens given hidden states

Is there a way to let HF generate tokens based on the last hidden states?