michael/stable-diffusion-webui
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages 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 module_in_gpu = None
cpu = torch.device("cpu") cpu = torch.device("cpu")
gpu = torch.device("cuda") if torch.has_cuda:
device = gpu if torch.cuda.is_available() else cpu 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): def setup_for_low_vram(sd_model, use_medvram):
parents = {} parents = {}