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
- MODELS = {
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