RamAnanth1 commited on
Commit
c0af429
1 Parent(s): cc7953e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -9
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
- # 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(
 
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(