Hugging Face's logo

Spaces:
xywwww
/
scene_diffusion___
Runtime error

App Files Community
1
xywwww commited on
Commit
524ecf2
·
verified ·
1 Parent(s): 62139d5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -20,7 +20,7 @@ apply_canny = CannyDetector()
20
 
21
  model = create_model('./models/cldm_v21_512_latctrl_coltrans.yaml').cpu()
22
  # /checkpoints/epoch=25-step=112553.ckpt
23
- model.load_state_dict(load_state_dict('xywwww/scene_diffusion', location='cuda'),strict=False)
24
  model = model.cuda()
25
  ddim_sampler = DDIMSampler(model)
26
 
 
20
 
21
  model = create_model('./models/cldm_v21_512_latctrl_coltrans.yaml').cpu()
22
  # /checkpoints/epoch=25-step=112553.ckpt
23
+ model.load_state_dict(load_state_dict('xywwww/scene_diffusion/checkpoints/epoch=25-step=112553.ckpt', location='cuda'),strict=False)
24
  model = model.cuda()
25
  ddim_sampler = DDIMSampler(model)
26