From 01976a9908e6a8fd0906c8bcc2fc3488e45f927f Mon Sep 17 00:00:00 2001
From: Param Bole <parambole@google.com>
Date: Thu, 25 Jul 2024 00:03:49 +0000
Subject: [PATCH] point change

---
 docker_maxdiffusion_image_upload.sh | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/docker_maxdiffusion_image_upload.sh b/docker_maxdiffusion_image_upload.sh
index 7b358cc7..1355dcc8 100644
--- a/docker_maxdiffusion_image_upload.sh
+++ b/docker_maxdiffusion_image_upload.sh
@@ -62,6 +62,8 @@ fi
 # Default: Don't delete local image
 DELETE_LOCAL_IMAGE="${DELETE_LOCAL_IMAGE:-false}"
 
+gcloud auth configure-docker us-docker.pkg.dev --quiet
+
 COMMIT_HASH=$(git rev-parse --short HEAD)
 
 echo "Building JAX Stable Stack MaxDiffusion at commit hash ${COMMIT_HASH} . . ."