EPiC-LowRes / download /download.py
roll-ai's picture
Update download/download.py
5bf643c verified
import os
os.environ["HF_HUB_DISABLE_XET"] = "1"
os.environ["HF_HUB_ENABLE_HF_TRANSFER"] = "0"
from huggingface_hub import snapshot_download
def download_model():
snapshot_download(
repo_id="tencent/DepthCrafter", # 4 GB
local_dir="/app/pretrained/DepthCrafter",
resume_download=True,
max_workers=1,
force_download=True,
)
snapshot_download(
repo_id="stabilityai/stable-video-diffusion-img2vid", # 25 gb
local_dir="/app/pretrained/stable-video-diffusion-img2vid",
resume_download=True,
max_workers=1,
force_download=True,
)
snapshot_download(
repo_id= "Qwen/Qwen2.5-VL-7B-Instruct", # 16 GB
local_dir="/app/pretrained/Qwen2.5-VL-7B-Instruct",
resume_download=True,
max_workers=1,
force_download=True,
)
snapshot_download(
repo_id="THUDM/CogVideoX-5b-I2V", # 22 GB
local_dir="/app/pretrained/CogVideoX-5b-I2V",
resume_download=True,
max_workers=1,
force_download=True,
)
download_model()