Mirroristas/whisper
2
0
Fork 0
mirror of https://github.com/openai/whisper.git synced 2025-11-26 23:46:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

add another hasattr check for torch.backends.mps

Browse Source
This commit is contained in:
Jong Wook Kim 2023-01-18 14:44:02 -08:00 committed by GitHub
parent 4b77a81c1f
commit 51c785f7c9
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
whisper/__init__.py
View File

@ -94,7 +94,7 @@ def load_model(name: str, device: Optional[Union[str, torch.device]] = None, dow
if device is None: if device is None:
if torch.cuda.is_available(): if torch.cuda.is_available():
device = "cuda" device = "cuda"
elif torch.backends.mps.is_available(): elif hasattr(torch.backends, "mps") and torch.backends.mps.is_available():
device = "mps" device = "mps"
else: else:
device = "cpu" device = "cpu"