The hidden states are passed as inputs to a model head to produce the final output.