riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-08-01 02:31:11 +06:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

Check if the current compiled version of pytorch supports MPS (#30664)

Browse Source
This commit is contained in:
jiaqianjing 2024-05-06 16:32:19 +08:00 committed by GitHub
parent 307f632bb2
commit 09edd77f64
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/transformers/utils/import_utils.py
View File

@ -373,7 +373,7 @@ def is_torch_mps_available():
import torch
if hasattr(torch.backends, "mps"):
return torch.backends.mps.is_available()
return torch.backends.mps.is_available() and torch.backends.mps.is_built()
return False