first commit

annotator/uniformer_base/__init__.py

@@ -13,8 +13,8 @@
 
 
 import os
-from annotator.uniformer.mmseg.apis import init_segmentor, inference_segmentor, show_result_pyplot
-from annotator.uniformer.mmseg.core.evaluation import get_palette
+from annotator.uniformer_base.mmseg.apis import init_segmentor, inference_segmentor, show_result_pyplot
+from annotator.uniformer_base.mmseg.core.evaluation import get_palette
 from annotator.util import annotator_ckpts_path
 
 import pdb

app.py

@@ -77,7 +77,8 @@ def inpainting(img, res, rand_h, rand_h_1, rand_w, rand_w_1):
 
 
 model = create_model('./models/cldm_v15_unicontrol.yaml').cpu()
-model.load_state_dict(load_state_dict('./ckpts/unicontrol.ckpt', location='cuda'), strict=False)
+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)