Update devices.py

This commit is contained in:
源文雨 2022-11-12 11:09:28 +08:00 committed by GitHub
parent 76ab31e188
commit 1130d5df66
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -6,7 +6,7 @@ from modules import errors
# has_mps is only available in nightly pytorch (for now) and MasOS 12.3+.
# check `getattr` and try it for compatibility
def has_mps() -> bool:
if getattr(torch, 'has_mps', False): return False
if not getattr(torch, 'has_mps', False): return False
try:
torch.zeros(1).to(torch.device("mps"))
return True