nodes/ltdrdata-ComfyUI-Impact-Subpack
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Restored the default model download feature during the installation step.

Browse Source
This commit is contained in:
Dr.Lt.Data
2024-09-22 07:20:11 +09:00
parent 81b1a47987
commit 748dfcf830
1 changed files with 17 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
install.py
+17 -4
View File
@@ -19,8 +19,21 @@ if model_path is None:
ultralytics_bbox_path = os.path.join(model_path, "ultralytics", "bbox")
ultralytics_segm_path = os.path.join(model_path, "ultralytics", "segm")
if not os.path.exists(ultralytics_bbox_path):
os.makedirs(ultralytics_bbox_path)
if not os.path.exists(os.path.join(subpack_path, '..', '..', 'skip_download_model')):
if not os.path.exists(ultralytics_bbox_path):
os.makedirs(ultralytics_bbox_path)
if not os.path.exists(ultralytics_segm_path):
os.makedirs(ultralytics_segm_path)
if not os.path.exists(ultralytics_segm_path):
os.makedirs(ultralytics_segm_path)
if not os.path.exists(os.path.join(ultralytics_bbox_path, "face_yolov8m.pt")):
download_url("https://huggingface.co/Bingsu/adetailer/resolve/main/face_yolov8m.pt",
ultralytics_bbox_path)
if not os.path.exists(os.path.join(ultralytics_bbox_path, "hand_yolov8s.pt")):
download_url("https://huggingface.co/Bingsu/adetailer/resolve/main/hand_yolov8s.pt",
ultralytics_bbox_path)
if not os.path.exists(os.path.join(ultralytics_segm_path, "person_yolov8m-seg.pt")):
download_url("https://huggingface.co/Bingsu/adetailer/resolve/main/person_yolov8m-seg.pt",
ultralytics_segm_path)