jaidaken
f09734b0ee
Some checks failed
Python Linting / Run Ruff (push) Has been cancelled
Python Linting / Run Pylint (push) Has been cancelled
Full Comfy CI Workflow Runs / test-stable (12.1, , linux, 3.10, [self-hosted Linux], stable) (push) Has been cancelled
Full Comfy CI Workflow Runs / test-stable (12.1, , linux, 3.11, [self-hosted Linux], stable) (push) Has been cancelled
Full Comfy CI Workflow Runs / test-stable (12.1, , linux, 3.12, [self-hosted Linux], stable) (push) Has been cancelled
Full Comfy CI Workflow Runs / test-unix-nightly (12.1, , linux, 3.11, [self-hosted Linux], nightly) (push) Has been cancelled
Execution Tests / test (macos-latest) (push) Has been cancelled
Execution Tests / test (ubuntu-latest) (push) Has been cancelled
Execution Tests / test (windows-latest) (push) Has been cancelled
Test server launches without errors / test (push) Has been cancelled
Unit Tests / test (macos-latest) (push) Has been cancelled
Unit Tests / test (ubuntu-latest) (push) Has been cancelled
Unit Tests / test (windows-2022) (push) Has been cancelled
Includes 30 custom nodes committed directly, 7 Civitai-exclusive loras stored via Git LFS, and a setup script that installs all dependencies and downloads HuggingFace-hosted models on vast.ai. Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
42 lines
1.8 KiB
Python
42 lines
1.8 KiB
Python
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)
|