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

Merge pull request #3 from 920232796/master

Browse Source 
fix device support for mps
update the support for SD2.0
This commit is contained in:
Zac Liu
2022-12-06 09:17:57 +08:00
committed by GitHub
parent 3ebf977a6e 4929503258
commit a25dfebeed
4 changed files with 75 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
modules/devices.py
View File

@@ -38,8 +38,8 @@ def get_optimal_device():
if torch.cuda.is_available():
return torch.device(get_cuda_device_string())
# if has_mps():
# return torch.device("mps")
if has_mps():
return torch.device("mps")
return cpu