Option to set 'non_blocking' for to(device) in BatchEncoding and BatchFeature#34883

Merged

ArthurZucker merged 5 commits intohuggingface:mainfrom daniel-bogdoll:batch-encoding-with-non_blocking-option

Dec 9, 2024

+10-4

Commits on Nov 22, 2024

Option to set 'non_blocking' for to(device) operation for performance improvements. Defaults to 'false', thus no behavioral changes.
Daniel Bogdoll
committed
Enabling non_blocking in to() operation of BatchFeature.
Daniel Bogdoll
committed
Improved docstring on utilization of non_blocking
Daniel Bogdoll
committed

Commits on Nov 25, 2024

Force non_blocking as keyword argument

daniel-bogdoll
and
qubvel
authored

Commits on Dec 2, 2024

Merge branch 'main' into batch-encoding-with-non_blocking-option
daniel-bogdoll
authored