RamAnanth1 commited on
Commit
02583a6
1 Parent(s): 487f89b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -23,7 +23,7 @@ pose_checkpoint = "models/control_sd15_openpose.pth"
23
  canny_model = create_model('./models/cldm_v15.yaml').cpu()
24
  canny_model.load_state_dict(load_state_dict(cached_download(
25
  hf_hub_url(REPO_ID, canny_checkpoint)
26
- ), location='cuda'))
27
  canny_model = canny_model.cuda()
28
  ddim_sampler = DDIMSampler(canny_model)
29
 
 
23
  canny_model = create_model('./models/cldm_v15.yaml').cpu()
24
  canny_model.load_state_dict(load_state_dict(cached_download(
25
  hf_hub_url(REPO_ID, canny_checkpoint)
26
+ ), location='cpu'))
27
  canny_model = canny_model.cuda()
28
  ddim_sampler = DDIMSampler(canny_model)
29