#!/usr/bin/env bash
# vast.ai onstart: holt dieses Repo und startet das Setup.
set -euo pipefail
exec > >(tee -a /workspace/onstart.log) 2>&1
echo "=== onstart $(date -u) ==="

: "${HF_TOKEN:?HF_TOKEN fehlt - im Template unter Environment setzen}"
export HF_HUB_ENABLE_HF_TRANSFER=1
pip install -q --upgrade "huggingface_hub[hf_transfer]"

python - <<'PY'
import os
from huggingface_hub import snapshot_download
snapshot_download("Tirkaru/krea2-nvfp4-rig", local_dir="/workspace/rig",
                  allow_patterns=["scripts/*", "workflows/*"], token=os.environ["HF_TOKEN"])
print("Rig geladen nach /workspace/rig")
PY

chmod +x /workspace/rig/scripts/*.sh
exec bash /workspace/rig/scripts/setup.sh