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 |
|