Spaces:

xywwww
/

scene_diffusion___

Runtime error

xywwww commited on Dec 21, 2024

Commit

b30d66a

verified ·

1 Parent(s): b7aaaea

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -26,7 +26,7 @@ from huggingface_hub import hf_hub_download
 # model = model.cuda()
 # ddim_sampler = DDIMSampler(model)
 model = create_model('./models/cldm_v21_512_latctrl_coltrans.yaml')
-ckpt = hf_hub_download(repo_id="xywwww/scene_diffusion", filename="checkpoints/epoch=25-step=112553.ckpt")
 model.load_state_dict(load_state_dict(ckpt), strict=False)
 model = model.cuda()
 ddim_sampler = DDIMSampler(model)

 # model = model.cuda()
 # ddim_sampler = DDIMSampler(model)
 model = create_model('./models/cldm_v21_512_latctrl_coltrans.yaml')
+ckpt = hf_hub_download(repo_id="xywwww/scene_diffusion", filename="checkpoints/epoch=29-step=479999.ckpt")
 model.load_state_dict(load_state_dict(ckpt), strict=False)
 model = model.cuda()
 ddim_sampler = DDIMSampler(model)