Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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")
|