krea2-nvfp4-rig / scripts /onstart.sh
Tirkaru's picture
Upload folder using huggingface_hub
0695f8e verified
Raw
History Blame Contribute Delete
702 Bytes
#!/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