Aore commited on
Commit
3c6b545
·
1 Parent(s): 6c4669e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -1
app.py CHANGED
@@ -25,7 +25,6 @@ controlnet_path = "./models/sks_crack_controlnet.pth"
25
  model.load_state_dict(load_state_dict(sd_model_path, location='cpu'), strict = False)
26
  model.load_state_dict(load_state_dict(controlnet_path, location='cpu'), strict = False)
27
 
28
- model = model.cuda()
29
  model = model.to('cpu')
30
  ddim_sampler = DDIMSampler(model)
31
  init_mask = Image.open("379.png").convert("L")
 
25
  model.load_state_dict(load_state_dict(sd_model_path, location='cpu'), strict = False)
26
  model.load_state_dict(load_state_dict(controlnet_path, location='cpu'), strict = False)
27
 
 
28
  model = model.to('cpu')
29
  ddim_sampler = DDIMSampler(model)
30
  init_mask = Image.open("379.png").convert("L")