From ab37384bc90e63cc787a9dff2851daf72ad4d365 Mon Sep 17 00:00:00 2001
From: Evan Hosinski <ehosinski@krkn.tech>
Date: Thu, 22 Jan 2026 10:46:55 -0500
Subject: [PATCH] Guard the pull

---
 install.sh | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/install.sh b/install.sh
index bb806e8..628315d 100644
--- a/install.sh
+++ b/install.sh
@@ -119,6 +119,8 @@ IMAGE="${IMAGE_BASE}/${CUDA_VERSION}/tentacle:${IMAGE_TAG}"
 # -----------------------------
 # Registry login
 # -----------------------------
+docker logout "$REGISTRY" >/dev/null 2>&1 || true
+
 REG_USER="krkn-registry"
 
 read -rsp "Registry access token: " REG_PASS
@@ -147,7 +149,10 @@ WORKER_ID="${WORKER_ID:-$DEFAULT_WORKER_ID}"
 # Pull image
 # -----------------------------
 log "Pulling image: ${IMAGE}"
-docker pull "${IMAGE}"
+docker pull "${IMAGE}" || {
+  err "Failed to pull image. Check token permissions or image availability."
+  exit 1
+}
 
 # -----------------------------
 # Stop old container