postBuild

#!/bin/bash
set -e

export HF_TOKEN="${HF_TOKEN}"
export HF_HUB_ENABLE_HF_TRANSFER=1

python - <<'PY'
import os
from huggingface_hub import snapshot_download

repo_id = "stabilityai/stable-video-diffusion-img2vid-xt"

# Если есть /data — используем его (persistent storage). Если нет — /home/user.
root = "/data" if os.path.isdir("/data") else "/home/user"
local_dir = os.path.join(root, "models", "svd-xt")
os.makedirs(local_dir, exist_ok=True)

snapshot_download(
    repo_id=repo_id,
    local_dir=local_dir,
    local_dir_use_symlinks=False,
    resume_download=True,
    max_workers=4,
    token=os.environ.get("HF_TOKEN"),
)
print("✅ Downloaded to:", local_dir)
PY