aboutsummaryrefslogtreecommitdiffstats
path: root/modules/devices.py
diff options
context:
space:
mode:
authorzhaohu xing <920232796@qq.com>2022-11-29 02:28:41 +0000
committerzhaohu xing <920232796@qq.com>2022-11-29 02:28:41 +0000
commit75c4511e6b81ae8fb0dbd932043e8eb35cd09f72 (patch)
tree6f4662507be1d532a4e992f54f82d905fc450f3a /modules/devices.py
parent828438b4a190759807f9054932cae3a8b880ddf1 (diff)
downloadstable-diffusion-webui-gfx803-75c4511e6b81ae8fb0dbd932043e8eb35cd09f72.tar.gz
stable-diffusion-webui-gfx803-75c4511e6b81ae8fb0dbd932043e8eb35cd09f72.tar.bz2
stable-diffusion-webui-gfx803-75c4511e6b81ae8fb0dbd932043e8eb35cd09f72.zip
add AltDiffusion to webui
Signed-off-by: zhaohu xing <920232796@qq.com>
Diffstat (limited to 'modules/devices.py')
-rw-r--r--modules/devices.py4
1 files changed, 2 insertions, 2 deletions
diff --git a/modules/devices.py b/modules/devices.py
index 67165bf6..f30b6ebc 100644
--- a/modules/devices.py
+++ b/modules/devices.py
@@ -36,8 +36,8 @@ def get_optimal_device():
else:
return torch.device("cuda")
- if has_mps():
- return torch.device("mps")
+ # if has_mps():
+ # return torch.device("mps")
return cpu