Spaces:

Robert001
/

UniControl-Demo

Runtime error

Robert001 commited on Jul 25, 2023

Commit

9fcaa10

•

1 Parent(s): 316bfe4

first commit

Files changed (1) hide show

app.py CHANGED Viewed

@@ -77,7 +77,7 @@ def inpainting(img, res, rand_h, rand_h_1, rand_w, rand_w_1):
 model = create_model('./models/cldm_v15_unicontrol.yaml').cpu()
-model_path = 'https://huggingface.co/Robert001/UniControl-Model/blob/main/unicontrol_v1.1.ckpt'
 model.load_state_dict(load_state_dict(model_path, location='cuda'), strict=False)
 model = model.cuda()
 ddim_sampler = DDIMSampler(model)

 model = create_model('./models/cldm_v15_unicontrol.yaml').cpu()
+model_path = 'https://huggingface.co/Robert001/UniControl-Model/resolve/main/unicontrol_v1.1.ckpt'
 model.load_state_dict(load_state_dict(model_path, location='cuda'), strict=False)
 model = model.cuda()
 ddim_sampler = DDIMSampler(model)