RamAnanth1 commited on
Commit
570118a
1 Parent(s): 02583a6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -37,7 +37,7 @@ ddim_sampler_pose = DDIMSampler(pose_model)
37
  scribble_model = create_model('./models/cldm_v15.yaml').cpu()
38
  scribble_model.load_state_dict(load_state_dict(cached_download(
39
  hf_hub_url(REPO_ID, scribble_checkpoint)
40
- ), location='cuda'))
41
  scribble_model = canny_model.cuda()
42
  ddim_sampler_scribble = DDIMSampler(scribble_model)
43
 
 
37
  scribble_model = create_model('./models/cldm_v15.yaml').cpu()
38
  scribble_model.load_state_dict(load_state_dict(cached_download(
39
  hf_hub_url(REPO_ID, scribble_checkpoint)
40
+ ), location='cpu'))
41
  scribble_model = canny_model.cuda()
42
  ddim_sampler_scribble = DDIMSampler(scribble_model)
43