The .generate() function directly gives the output tokens. Is there a way to get the last hidden states of the decoder?
· Sign up or log in to comment