transformers

ccc08978 - Adding support for `device_map` directly in `pipeline(..)` function. (#17902)

Adding support for `device_map` directly in `pipeline(..)` function. (#17902) * Adding support for `device_map` directly in `pipeline(..)` function. * Updating the docstring. * Adding a better docstring * Put back type hints. * Blacked. (`make fixup` didn't work ??!!)