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}