get hidden state #1695
Description
In the process of inferring tokens, how to return the hidden state before the lm_head and the logits after the lm_head
{{ message }}
In the process of inferring tokens, how to return the hidden state before the lm_head and the logits after the lm_head