sentence-transformers
Fix #3185: Support `convert_to_numpy` for `output_value="token_embeddings"`
#3186
Open
Open
Fix #3185: Support `convert_to_numpy` for `output_value="token_embeddings"` #3186
bhavnicksm
wants to merge
3 commits into
huggingface:main
from
bhavnicksm:fix-3185
fix #3185: convert_to_numpy should be supported for token_embeddings
42c22349
fix #3185: shift embeddings to cpu first
616fcbe2
Add tests; remove dead code
2c228aa9
Reviewers
No reviews
Assignees
No one assigned
Labels
None yet
Milestone
No milestone
Login to write a write a comment.
Login via GitHub