Commit 1c6ab9e
authored
[utils] account for MPS when available in get_device(). (huggingface#11905)
* account for MPS when available in get_device().
* fix1 parent 265840a commit 1c6ab9e
1 file changed
+2
-0
lines changed| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
175 | 175 | | |
176 | 176 | | |
177 | 177 | | |
| 178 | + | |
| 179 | + | |
178 | 180 | | |
179 | 181 | | |
180 | 182 | | |
| |||
0 commit comments