Joseph Catrambone
commited on
Commit
Β·
6948da2
1
Parent(s):
497cb11
Rename controlnet_* to be consistent with ControlNet1.1 model naming scheme.
Browse files- control_mediapipe_face_sd15_v2_full.ckpt β control_v2p_sd15_mediapipe_face.full.ckpt +0 -0
- control_mediapipe_face_sd15_v2.pt β control_v2p_sd15_mediapipe_face.pth +0 -0
- control_mediapipe_face_sd15_v2.safetensors β control_v2p_sd15_mediapipe_face.safetensors +0 -0
- control_mediapipe_face_sd15_v2.yaml β control_v2p_sd15_mediapipe_face.yaml +0 -0
- control_mediapipe_face_sd21_v2_full.ckpt β control_v2p_sd21_mediapipe_face.full.ckpt +0 -0
- control_mediapipe_face_sd21_v2.pt β control_v2p_sd21_mediapipe_face.pth +0 -0
- control_mediapipe_face_sd21_v2.safetensors β control_v2p_sd21_mediapipe_face.safetensors +0 -0
- control_mediapipe_face_sd21_v2.yaml β control_v2p_sd21_mediapipe_face.yaml +0 -0
- gradio_face2image.py +2 -2
control_mediapipe_face_sd15_v2_full.ckpt β control_v2p_sd15_mediapipe_face.full.ckpt
RENAMED
|
File without changes
|
control_mediapipe_face_sd15_v2.pt β control_v2p_sd15_mediapipe_face.pth
RENAMED
|
File without changes
|
control_mediapipe_face_sd15_v2.safetensors β control_v2p_sd15_mediapipe_face.safetensors
RENAMED
|
File without changes
|
control_mediapipe_face_sd15_v2.yaml β control_v2p_sd15_mediapipe_face.yaml
RENAMED
|
File without changes
|
control_mediapipe_face_sd21_v2_full.ckpt β control_v2p_sd21_mediapipe_face.full.ckpt
RENAMED
|
File without changes
|
control_mediapipe_face_sd21_v2.pt β control_v2p_sd21_mediapipe_face.pth
RENAMED
|
File without changes
|
control_mediapipe_face_sd21_v2.safetensors β control_v2p_sd21_mediapipe_face.safetensors
RENAMED
|
File without changes
|
control_mediapipe_face_sd21_v2.yaml β control_v2p_sd21_mediapipe_face.yaml
RENAMED
|
File without changes
|
gradio_face2image.py
CHANGED
|
@@ -13,8 +13,8 @@ from laion_face_common import generate_annotation
|
|
| 13 |
from share import *
|
| 14 |
|
| 15 |
|
| 16 |
-
model = create_model('./
|
| 17 |
-
model.load_state_dict(load_state_dict('./
|
| 18 |
model = model.cuda()
|
| 19 |
ddim_sampler = DDIMSampler(model) # ControlNet _only_ works with DDIM.
|
| 20 |
|
|
|
|
| 13 |
from share import *
|
| 14 |
|
| 15 |
|
| 16 |
+
model = create_model('./control_v2p_sd21_mediapipe_face.yaml').cpu()
|
| 17 |
+
model.load_state_dict(load_state_dict('./control_v2p_sd21_mediapipe_face.full.ckpt', location='cuda'))
|
| 18 |
model = model.cuda()
|
| 19 |
ddim_sampler = DDIMSampler(model) # ControlNet _only_ works with DDIM.
|
| 20 |
|