transformers
8c00b53e - Adding `mps` as device for `Pipeline` class (#30080)

Commit
1 year ago
Adding `mps` as device for `Pipeline` class (#30080) * adding env variable for mps and is_torch_mps_available for Pipeline * fix linting errors * Remove environment overide Co-authored-by: amyeroberts <22614925+amyeroberts@users.noreply.github.com> --------- Co-authored-by: amyeroberts <22614925+amyeroberts@users.noreply.github.com>
Author
Parents
Loading