Update app.py

app.py

@@ -17,15 +17,15 @@ from cldm.ddim_hacked import DDIMSampler
 from PIL import Image
 
 # os.environ["no_proxy"] = "localhost,127.0.0.1,::1"
-device = "cuda:0"
+device = "cpu"
 
-model = create_model('./models/cldm_v15.yaml').cpu()
+model = create_model('./models/cldm_v15_cpu.yaml').cpu()
 sd_model_path = "./models/sks_crack_ppl.ckpt"
 controlnet_path = "./models/sks_crack_controlnet.pth"
 model.load_state_dict(load_state_dict(sd_model_path, location='cpu'), strict = False)
 model.load_state_dict(load_state_dict(controlnet_path, location='cpu'), strict = False)
 
-model = model.to('cpu')
+# model = model.cuda()
 ddim_sampler = DDIMSampler(model)
 init_mask = Image.open("379.png").convert("L")
 
@@ -43,7 +43,7 @@ def model_sample(mask,
         mask = resize_image(mask, 512)
         H, W, C= mask.shape
 
-        control = torch.from_numpy(mask.copy()).float().cuda() / 255.0
+        control = torch.from_numpy(mask.copy()).float().to(device) / 255.0
         control = torch.stack([control for _ in range(num_samples)], dim=0)
         control = einops.rearrange(control, 'b h w c -> b c h w').clone()