src/transformers/pipelines/base.py · 4144c354e9f54bde29357031fb3e48b23f60b836 · chenpangpang / transformers

auto-detect device when no device is passed to pipeline (#31398) · 4144c354

Fanli Lin authored Jun 19, 2024



* fix device

* Update src/transformers/pipelines/base.py
Co-authored-by: amyeroberts <22614925+amyeroberts@users.noreply.github.com>

* bug fix

* add warning

---------
Co-authored-by: amyeroberts <22614925+amyeroberts@users.noreply.github.com>

4144c354

base.py 56.1 KB

Replace base.py