File size: 3,057 Bytes
86f5cca |
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 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 |
import os
os.environ["HF_ENDPOINT"] = "https://hf-mirror.com"
from huggingface_hub import hf_hub_download
COMFYUI_PATH = r"/path/to/ComfyUI"
# ==============================================
# 官方 SD 权重
# ==============================================
hf_hub_download(
repo_id="stabilityai/stable-diffusion-xl-base-1.0",
filename="sd_xl_base_1.0.safetensors",
local_dir=rf"{COMFYUI_PATH}/models/checkpoints",
)
hf_hub_download(
repo_id="runwayml/stable-diffusion-v1-5",
filename="v1-5-pruned.safetensors",
local_dir=rf"{COMFYUI_PATH}/models/checkpoints",
)
# ==============================================
# AnimateDiff 权重
# ==============================================
for animatediff_model in ["mm_sd_v15_v2.ckpt", "mm_sdxl_v10_beta.ckpt", "v3_sd15_mm.ckpt"]:
hf_hub_download(
repo_id="guoyww/animatediff",
filename=animatediff_model,
local_dir=rf"{COMFYUI_PATH}/models/animatediff_models",
)
for animatediff_model in ["temporaldiff-v1-animatediff.safetensors"]:
hf_hub_download(
repo_id="CiaraRowles/TemporalDiff",
filename=animatediff_model,
local_dir=rf"{COMFYUI_PATH}/models/animatediff_models",
)
for lora_model in [
"v2_lora_PanLeft.ckpt",
"v2_lora_PanRight.ckpt",
"v2_lora_RollingAnticlockwise.ckpt",
"v2_lora_RollingClockwise.ckpt",
"v2_lora_TiltDown.ckpt",
"v2_lora_TiltUp.ckpt",
"v2_lora_ZoomIn.ckpt",
"v2_lora_ZoomOut.ckpt",
]:
hf_hub_download(
repo_id="guoyww/animatediff",
filename=lora_model,
local_dir=rf"{COMFYUI_PATH}/models/animatediff_motion_lora",
)
# ==============================================
# ControlNet 权重
# ==============================================
for controlnet_model in ["control_v11p_sd15_openpose.pth", "control_v11f1p_sd15_depth.pth", "control_v11p_sd15_seg.pth"]:
hf_hub_download(
repo_id="lllyasviel/ControlNet-v1-1",
filename=controlnet_model,
local_dir=rf"{COMFYUI_PATH}/models/controlnet",
)
# ==============================================
# SAM 权重
# ==============================================
for sam_model in ["groundingdino_swinb_cogcoor.pth", "GroundingDINO_SwinB.cfg.py"]:
hf_hub_download(
repo_id="ShilongLiu/GroundingDINO",
filename=sam_model,
local_dir=rf"{COMFYUI_PATH}/models/grounding-dino/",
)
# ==============================================
# IP-Adapter 权重
# ==============================================
for ip_adapter_model in ["models/ip-adapter-plus_sd15.safetensors"]:
hf_hub_download(
repo_id="h94/IP-Adapter",
filename=ip_adapter_model,
local_dir=rf"{COMFYUI_PATH}/models/ipadapter",
)
for ip_adapter_clip_model in ["models/image_encoder/model.safetensors"]:
hf_hub_download(
repo_id="h94/IP-Adapter",
filename=ip_adapter_clip_model,
local_dir=rf"{COMFYUI_PATH}/models/clip_vision/",
)
|