| # Download the released ROMA checkpoint (EurekaTian/ROMA, ~16-22GB bf16 safetensors) into the | |
| # location the demos expect by default: whole_model/model. | |
| # Run this INSIDE the container (working dir /app). Re-runs resume/skip already-downloaded files. | |
| set -euo pipefail | |
| REPO_ID="${ROMA_HF_REPO:-EurekaTian/ROMA}" | |
| TARGET_DIR="${ROMA_MODEL_PATH:-whole_model/model}" | |
| export HF_HUB_ENABLE_HF_TRANSFER="${HF_HUB_ENABLE_HF_TRANSFER:-1}" | |
| echo "Downloading ${REPO_ID} -> ${TARGET_DIR}" | |
| mkdir -p "${TARGET_DIR}" | |
| TOKEN_ARGS=() | |
| if [[ -n "${HF_TOKEN:-}" ]]; then | |
| TOKEN_ARGS=(--token "${HF_TOKEN}") | |
| fi | |
| huggingface-cli download "${REPO_ID}" \ | |
| --local-dir "${TARGET_DIR}" \ | |
| --local-dir-use-symlinks False \ | |
| "${TOKEN_ARGS[@]}" | |
| echo "Done. Checkpoint is at: ${TARGET_DIR}" | |
| ls -lh "${TARGET_DIR}" | head -40 | |