From 3d594bdd369b353a70570a1d6cf06d4fa700bf68 Mon Sep 17 00:00:00 2001
From: Param Bole <parambole@google.com>
Date: Thu, 25 Jul 2024 16:40:58 +0000
Subject: [PATCH] test

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

diff --git a/docker_maxdiffusion_image_upload.sh b/docker_maxdiffusion_image_upload.sh
index d5d756be..1dabb5ee 100644
--- a/docker_maxdiffusion_image_upload.sh
+++ b/docker_maxdiffusion_image_upload.sh
@@ -77,7 +77,7 @@ docker build --no-cache \
   --build-arg COMMIT_HASH=${COMMIT_HASH} \
   --build-arg MAXDIFFUSION_REQUIREMENTS_FILE=${MAXDIFFUSION_REQUIREMENTS_FILE} \
   --network=host \
-  -t us-docker.pkg.dev/${PROJECT_ID}/${CLOUD_IMAGE_NAME}/tpu:${FULL_IMAGE_TAG} \
+  -t gcr.io/${PROJECT_ID}/${CLOUD_IMAGE_NAME}/tpu:${FULL_IMAGE_TAG} \
   -f maxdiffusion_jax_stable_stack_tpu.Dockerfile .
 
 docker push gcr.io/${PROJECT_ID}/${CLOUD_IMAGE_NAME}/tpu:${FULL_IMAGE_TAG}