transformers
Option to set 'non_blocking' for to(device) in BatchEncoding and BatchFeature
#34883
Merged
Merged
Option to set 'non_blocking' for to(device) in BatchEncoding and BatchFeature #34883
ArthurZucker
merged
5 commits into
huggingface:main
from
daniel-bogdoll:batch-encoding-with-non_blocking-option
Option to set 'non_blocking' for to(device) operation for performance…
88eddf86
qubvel
commented
on 2024-11-22
Enabling non_blocking in to() operation of BatchFeature.
08f5d4b3
daniel-bogdoll
changed the title
Improved docstring on utilization of non_blocking
9f465d7c
qubvel
approved these changes
on 2024-11-25
Force non_blocking as keyword argument
4e8e786d
Merge branch 'main' into batch-encoding-with-non_blocking-option
2d2fe861
ArthurZucker
commented
on 2024-12-05
ArthurZucker
approved these changes
on 2024-12-09
Reviewers
Assignees
No one assigned
Labels
None yet
Milestone
No milestone
Login to write a write a comment.
Login via GitHub