jhj0517
commited on
Commit
·
f2a34e7
1
Parent(s):
cb528cc
Add face detect model
Browse files
modules/live_portrait/model_downloader.py
CHANGED
@@ -1,12 +1,13 @@
|
|
1 |
import requests
|
2 |
from tqdm import tqdm
|
3 |
|
4 |
-
|
5 |
"appearance_feature_extractor": "https://huggingface.co/Kijai/LivePortrait_safetensors/resolve/main/appearance_feature_extractor.safetensors",
|
6 |
"motion_extractor": "https://huggingface.co/Kijai/LivePortrait_safetensors/blob/main/motion_extractor.safetensors",
|
7 |
"warping_module": "https://huggingface.co/Kijai/LivePortrait_safetensors/tree/main/warping_module.safetensors",
|
8 |
"spade_generator ": "https://huggingface.co/Kijai/LivePortrait_safetensors/tree/main/spade_generator.safetensors ",
|
9 |
"stitching_retargeting_module ": "https://huggingface.co/Kijai/LivePortrait_safetensors/tree/main/stitching_retargeting_module.safetensors",
|
|
|
10 |
}
|
11 |
|
12 |
|
|
|
1 |
import requests
|
2 |
from tqdm import tqdm
|
3 |
|
4 |
+
MODELS_URL = {
|
5 |
"appearance_feature_extractor": "https://huggingface.co/Kijai/LivePortrait_safetensors/resolve/main/appearance_feature_extractor.safetensors",
|
6 |
"motion_extractor": "https://huggingface.co/Kijai/LivePortrait_safetensors/blob/main/motion_extractor.safetensors",
|
7 |
"warping_module": "https://huggingface.co/Kijai/LivePortrait_safetensors/tree/main/warping_module.safetensors",
|
8 |
"spade_generator ": "https://huggingface.co/Kijai/LivePortrait_safetensors/tree/main/spade_generator.safetensors ",
|
9 |
"stitching_retargeting_module ": "https://huggingface.co/Kijai/LivePortrait_safetensors/tree/main/stitching_retargeting_module.safetensors",
|
10 |
+
"face_yolov8n": "https://huggingface.co/Bingsu/adetailer/resolve/main/face_yolov8n.pt"
|
11 |
}
|
12 |
|
13 |
|
modules/utils/paths.py
CHANGED
@@ -11,7 +11,8 @@ MODEL_PATHS = {
|
|
11 |
"motion_extractor": os.path.join(MODELS_DIR, "motion_extractor.safetensors"),
|
12 |
"warping_module": os.path.join(MODELS_DIR, "warping_module.safetensors"),
|
13 |
"spade_generator": os.path.join(MODELS_DIR, "spade_generator.safetensors"),
|
14 |
-
"stitching_retargeting_module": os.path.join(MODELS_DIR, "stitching_retargeting_module.safetensors")
|
|
|
15 |
}
|
16 |
|
17 |
|
|
|
11 |
"motion_extractor": os.path.join(MODELS_DIR, "motion_extractor.safetensors"),
|
12 |
"warping_module": os.path.join(MODELS_DIR, "warping_module.safetensors"),
|
13 |
"spade_generator": os.path.join(MODELS_DIR, "spade_generator.safetensors"),
|
14 |
+
"stitching_retargeting_module": os.path.join(MODELS_DIR, "stitching_retargeting_module.safetensors"),
|
15 |
+
"face_yolov8n": os.path.join(MODELS_DIR, "face_yolov8n.pt")
|
16 |
}
|
17 |
|
18 |
|