Spaces:

Aore
/

controlNet-seg2image

Runtime error

Aore commited on May 22, 2023

Commit

11efc35

1 Parent(s): 2bd9f65

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -20,8 +20,8 @@ from PIL import Image
 device = "cuda:0"
 model = create_model('./models/cldm_v15.yaml').cpu()
-sd_model_path = ".models/sks_crack_ppl.ckpt"
-controlnet_path = ".models/sks_crack_controlnet.pth"
 model.load_state_dict(load_state_dict(sd_model_path, location='cpu'), strict = False)
 model.load_state_dict(load_state_dict(controlnet_path, location='cpu'), strict = False)

 device = "cuda:0"
 model = create_model('./models/cldm_v15.yaml').cpu()
+sd_model_path = "./models/sks_crack_ppl.ckpt"
+controlnet_path = "./models/sks_crack_controlnet.pth"
 model.load_state_dict(load_state_dict(sd_model_path, location='cpu'), strict = False)
 model.load_state_dict(load_state_dict(controlnet_path, location='cpu'), strict = False)