transformers
9d4a4550 - `pipeline` support for `device="mps"` (or any other string) (#18494)
Commit
View On GitHub
Commit
3 years ago
`pipeline` support for `device="mps"` (or any other string) (#18494) * `pipeline` support for `device="mps"` (or any other string) * Simplify `if` nesting * Update src/transformers/pipelines/base.py Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com> * Fix? @sgugger * passing `attr=None` is not the same as not passing `attr` 🤯 Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com>
Author
julien-c julien-c
Parents
0d0aada5
FAQ Terms Privacy Refunds Impressum
Loading