How to get last hidden states?

#54
by ZappY-AI - opened

The .generate() function gives out the token ids but not the hidden states even if we give 'output_hidden_states=True' as argument.

Sign up or log in to comment