Spaces:

otmanheddouch
/

house_design

Runtime error

otmanheddouch commited on May 14, 2023

Commit

0e1a8b1

1 Parent(s): c6647f4

Update gradio_canny2image.py

Files changed (1) hide show

gradio_canny2image.py CHANGED Viewed

@@ -17,8 +17,8 @@ from cldm.ddim_hacked import DDIMSampler
 apply_canny = CannyDetector()
-model = create_model('./models/cldm_v15.yaml').cpu()
-model.load_state_dict(load_state_dict('./models/control_sd15_canny.pth', location='cuda'))
 model = model.cuda()
 ddim_sampler = DDIMSampler(model)

 apply_canny = CannyDetector()
+model = create_model('models/cldm_v15.yaml').cpu()
+model.load_state_dict(load_state_dict('models/control_sd15_canny.pth', location='cuda'))
 model = model.cuda()
 ddim_sampler = DDIMSampler(model)