Spaces:

xywwww
/

scene_diffusion___

Runtime error

xywwww commited on Dec 12, 2024

Commit

62139d5

verified ·

1 Parent(s): 9c9e7cd

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -19,7 +19,8 @@ from ldm.models.diffusion.ddpm import LatentDiffusion
 apply_canny = CannyDetector()
 model = create_model('./models/cldm_v21_512_latctrl_coltrans.yaml').cpu()
-model.load_state_dict(load_state_dict('xywwww/scene_diffusion/checkpoints/epoch=25-step=112553.ckpt', location='cuda'),strict=False)
 model = model.cuda()
 ddim_sampler = DDIMSampler(model)

 apply_canny = CannyDetector()
 model = create_model('./models/cldm_v21_512_latctrl_coltrans.yaml').cpu()
+# /checkpoints/epoch=25-step=112553.ckpt
+model.load_state_dict(load_state_dict('xywwww/scene_diffusion', location='cuda'),strict=False)
 model = model.cuda()
 ddim_sampler = DDIMSampler(model)