Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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()
|