Skip to content

GitLab

"README_ORIGIN.md" did not exist on "eda1a7cad3b58d33353ce8f4145c66237a65c238"
Switch branch/tag
Find file Normal viewHistoryPermalink
install.py 1.78 KB
Newer Older
wuxk1's avatar
optim for test1,2,5  
wuxk1 committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41 
import os
import sys
from torchvision.datasets.utils import download_url
import logging

subpack_path = os.path.join(os.path.dirname(__file__))

comfy_path = os.environ.get('COMFYUI_PATH')
if comfy_path is None:
    comfy_path = os.path.abspath(os.path.join(os.path.dirname(__file__), '..', '..'))
    logging.warning(f"\nWARN: The `COMFYUI_PATH` environment variable is not set. Assuming `{comfy_path}` as the ComfyUI path.")

sys.path.append(comfy_path)

model_path = os.environ.get('COMFYUI_MODEL_PATH')
if model_path is None:
    model_path = os.path.abspath(os.path.join(comfy_path, 'models'))
    logging.warning(f"\nWARN: The `COMFYUI_MODEL_PATH` environment variable is not set. Assuming `{model_path}` as the ComfyUI path.")


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