nodes/thecooltechguy-ComfyUI-Stable-Video-Diffusion
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

added prestartup script to auto-download stable video diffusion models if user doesn't already have them locally

Browse Source
This commit is contained in:
thecooltechguy
2023-11-24 04:46:40 +00:00
parent 8cf1521fa9
commit 1d3917cfbf
1 changed files with 20 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
prestartup_script.py
+20
View File
@@ -0,0 +1,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 {model_name}...")
os.system(f'wget -O {os.path.join(svd_models_dir, model_name)} "{model_url}"')