RamAnanth1 commited on
Commit
daf30b7
1 Parent(s): 68e52a7

Update app.py

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