Add mps support to big inference modeling (#1545)
* Add mps support
* make style
* Fix syntax
Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com>
* Fix condition
Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com>
---------
Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com>