transformers
Adding support for `device_map` directly in `pipeline(..)` function.
#17902
Merged
Merged
Adding support for `device_map` directly in `pipeline(..)` function. #17902
Narsil
merged
5 commits into
huggingface:main
from
Narsil:main
younesbelkada
approved these changes
on 2022-06-27
Adding support for `device_map` directly in `pipeline(..)` function.
53313a17
Updating the docstring.
d325853f
Adding a better docstring
b7eb60c6
Put back type hints.
2ba913fe
Blacked. (`make fixup` didn't work ??!!)
d6e64d3c
Login to write a write a comment.
Login via GitHub