Update start.sh

start.sh

@@ -5,11 +5,21 @@ set -euo pipefail
 
 cd /workspace
 
+# ── Authenticate with Hugging Face (for gated models like Llama-3) ──
+if [ -n "${HF_TOKEN:-}" ]; then
+  echo "Logging in to Hugging Face Hub ..."
+  huggingface-cli login --token "$HF_TOKEN" --add-to-git-credential 2>/dev/null || true
+else
+  echo "WARNING: HF_TOKEN is not set. Gated models (e.g. Meta-Llama-3) will fail to download." >&2
+fi
+
 # pre download ckpts
 python - <<'PY'
 from huggingface_hub import snapshot_download
 snapshot_download("nvidia/Kimodo-SOMA-RP-v1")
 snapshot_download("nvidia/Kimodo-G1-RP-v1")
+# LLM2Vec text-encoder requires this gated model
+snapshot_download("meta-llama/Meta-Llama-3-8B-Instruct")
 print("Checkpoint download complete.")
 PY
 
@@ -42,4 +52,4 @@ done
 
 # launching the demo
 echo "Starting demo on :7860 ..."
-exec kimodo_demo
+exec kimodo_demo