Aore commited on
Commit
18da007
·
1 Parent(s): da07e2d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -4
app.py CHANGED
@@ -23,11 +23,7 @@ apply_uniformer = UniformerDetector()
23
 
24
  model = create_model('./models/cldm_v15.yaml').cpu()
25
 
26
- # ckpt for sd 1.5 DB finetuned with crack 500
27
- resume_path ="./models/control_sks_crack_ppl.ckpt"
28
- # ckpt for controlnet with sd 1.5 weights ( strict = False ), finetuned with ADE20K controlnet weight by crack500
29
  controlnet_path = "./models/sks_crack500_epoch_19.ckpt"
30
- model.load_state_dict(load_state_dict(resume_path, location='cpu'))
31
  model.load_state_dict(load_state_dict(controlnet_path, location='cpu'), strict = False)
32
 
33
  model = model.cuda()
 
23
 
24
  model = create_model('./models/cldm_v15.yaml').cpu()
25
 
 
 
 
26
  controlnet_path = "./models/sks_crack500_epoch_19.ckpt"
 
27
  model.load_state_dict(load_state_dict(controlnet_path, location='cpu'), strict = False)
28
 
29
  model = model.cuda()