hoelee/stable-diffusion-webui
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2025-08-09 05:39:47 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update devices.py

Browse Source
This commit is contained in:
源文雨
2022-11-12 11:09:28 +08:00
committed by GitHub
parent 76ab31e188
commit 1130d5df66
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
modules/devices.py
View File

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