diff --git a/prestartup_script.py b/prestartup_script.py new file mode 100644 index 0000000..68f899c --- /dev/null +++ b/prestartup_script.py @@ -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}"') \ No newline at end of file