Spaces:
Runtime error
Runtime error
Commit
·
c0af429
1
Parent(s):
cc7953e
Update app.py
Browse files
app.py
CHANGED
|
@@ -15,15 +15,15 @@ from cldm.model import create_model, load_state_dict
|
|
| 15 |
|
| 16 |
from huggingface_hub import hf_hub_url, cached_download
|
| 17 |
|
| 18 |
-
|
| 19 |
-
|
| 20 |
-
|
| 21 |
-
|
| 22 |
-
|
| 23 |
-
REPO_ID = "webui/ControlNet-modules-safetensors"
|
| 24 |
-
canny_checkpoint = "control_canny-fp16.safetensors"
|
| 25 |
-
scribble_checkpoint = "control_scribble-fp16.safetensors"
|
| 26 |
-
pose_checkpoint = "control_openpose-fp16.safetensors"
|
| 27 |
|
| 28 |
canny_model = create_model('./models/cldm_v15.yaml').cpu()
|
| 29 |
canny_model.load_state_dict(load_state_dict(cached_download(
|
|
|
|
| 15 |
|
| 16 |
from huggingface_hub import hf_hub_url, cached_download
|
| 17 |
|
| 18 |
+
REPO_ID = "lllyasviel/ControlNet"
|
| 19 |
+
canny_checkpoint = "models/control_sd15_canny.pth"
|
| 20 |
+
scribble_checkpoint = "models/control_sd15_scribble.pth"
|
| 21 |
+
pose_checkpoint = "models/control_sd15_openpose.pth"
|
| 22 |
+
|
| 23 |
+
# REPO_ID = "webui/ControlNet-modules-safetensors"
|
| 24 |
+
# canny_checkpoint = "control_canny-fp16.safetensors"
|
| 25 |
+
# scribble_checkpoint = "control_scribble-fp16.safetensors"
|
| 26 |
+
# pose_checkpoint = "control_openpose-fp16.safetensors"
|
| 27 |
|
| 28 |
canny_model = create_model('./models/cldm_v15.yaml').cpu()
|
| 29 |
canny_model.load_state_dict(load_state_dict(cached_download(
|