setup/download_cogvideo_weights.py

from huggingface_hub import snapshot_download

# Download the entire model repository and store it locally
model_path = snapshot_download(repo_id="THUDM/CogVideoX-2b", cache_dir="./CogVideoX-2b")

print(f"Model downloaded to: {model_path}")