sentence-transformers
Added move_to_cpu flag in encode method to avoid OOM for token embeddings.
#1812
Open
Open
Added move_to_cpu flag in encode method to avoid OOM for token embeddings. #1812
bodin-e
wants to merge
2 commits into
huggingface:main
from
bodin-e:move-to-cpu-flag-in-encode
Added move_to_cpu flag in encode method to avoid OOM for token embedd…
9b9b4de8
Updated docstring
0489f237
Reviewers
No reviews
Assignees
No one assigned
Labels
None yet
Milestone
No milestone
Login to write a write a comment.
Login via GitHub