-
Notifications
You must be signed in to change notification settings - Fork 25
/
prestartup_script.py
20 lines (16 loc) · 1.07 KB
/
prestartup_script.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
import os
import folder_paths
import sys
svd_models_dir = os.path.join(folder_paths.models_dir, "svd")
os.makedirs(svd_models_dir, exist_ok=True)
SVD_MODEL_URLS = {
"svd.safetensors" : "https://huggingface.co/stabilityai/stable-video-diffusion-img2vid/resolve/main/svd.safetensors?download=true",
"svd_image_decoder.safetensors" : "https://huggingface.co/stabilityai/stable-video-diffusion-img2vid/resolve/main/svd_image_decoder.safetensors?download=true",
"svd_xt.safetensors" : "https://huggingface.co/stabilityai/stable-video-diffusion-img2vid-xt/resolve/main/svd_xt.safetensors?download=true",
"svd_xt_image_decoder.safetensors" : "https://huggingface.co/stabilityai/stable-video-diffusion-img2vid-xt/resolve/main/svd_xt_image_decoder.safetensors?download=true"
}
existing_svd_models = os.listdir(svd_models_dir)
for model_name, model_url in SVD_MODEL_URLS.items():
if model_name not in existing_svd_models:
print(f"Downloading SVD model: {model_name}...")
os.system(f'wget -q --show-progress -O {os.path.join(svd_models_dir, model_name)} "{model_url}"')