Hugging Face's logo

Spaces:
RamAnanth1
/
ControlNet
Runtime error

App Files Community
12
RamAnanth1 commited on
Commit
4f291da
·
1 Parent(s): b3ba9f7

Bug fix

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -43,7 +43,7 @@ scribble_model = create_model('./models/cldm_v15.yaml').cpu()
43
  scribble_model.load_state_dict(load_state_dict(cached_download(
44
  hf_hub_url(REPO_ID, scribble_checkpoint)
45
  ), location='cpu'))
46
- scribble_model = canny_model.cuda()
47
  ddim_sampler_scribble = DDIMSampler(scribble_model)
48
 
49
  save_memory = False
 
43
  scribble_model.load_state_dict(load_state_dict(cached_download(
44
  hf_hub_url(REPO_ID, scribble_checkpoint)
45
  ), location='cpu'))
46
+ scribble_model = scribble_model.cuda()
47
  ddim_sampler_scribble = DDIMSampler(scribble_model)
48
 
49
  save_memory = False