Returning all hidden states
#97
by
carolinmschuster
- opened
Hi there,
currently the model is only returning the last hidden state when loaded with model = AutoModel.from_pretrained("jinaai/jina-embeddings-v3", output_hidden_states=True). For research purposes it would be very helpful if returning all hidden states was also supported for this model.
Best regards!
Ok I implemented this function, if you want you can take a look if this is what you wanted. We will probably merge it soon: https://huggingface.co/jinaai/xlm-roberta-flash-implementation/discussions/56
michael-guenther
changed discussion status to
closed