mirror/stable-diffusion-webui
2
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2024-11-21 03:11:40 +08:00
Code Issues Packages Projects Releases Wiki Activity

Miss device type for option --medvram

Browse Source
This commit is contained in:
xeonvs 2022-09-07 18:09:30 +02:00
parent 65fbefd033
commit aaeeef82fa
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
modules/lowvram.py
View File

@ -2,9 +2,12 @@ import torch
module_in_gpu = None
cpu = torch.device("cpu")
gpu = torch.device("cuda")
device = gpu if torch.cuda.is_available() else cpu
if torch.has_cuda:
device = gpu = torch.device("cuda")
elif torch.has_mps:
device = gpu = torch.device("mps")
else:
device = gpu = torch.device("cpu")
def setup_for_low_vram(sd_model, use_medvram):
parents = {}