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()