diff --git a/.github/workflows/UploadDockerImages.yml b/.github/workflows/UploadDockerImages.yml index 45a56c20e..ba1e7b769 100644 --- a/.github/workflows/UploadDockerImages.yml +++ b/.github/workflows/UploadDockerImages.yml @@ -68,7 +68,7 @@ jobs: - uses: actions/checkout@v3 - name: build jax stable image run : | - project=tpu-prod-env-multipod + project=supercomputer-testing local_image_name=maxtext_gpu_local_jax_stable cloud_image_name=maxtext_gpu_jax_stable bash docker_build_dependency_image.sh LOCAL_IMAGE_NAME=$local_image_name MODE=stable DEVICE=gpu @@ -81,7 +81,7 @@ jobs: docker push gcr.io/$project/${cloud_image_name}:${image_date} - name: build jax nightly image run : | - project=tpu-prod-env-multipod + project=supercomputer-testing local_image_name=maxtext_gpu_local_jax_nightly cloud_image_name=maxtext_gpu_jax_nightly bash docker_build_dependency_image.sh LOCAL_IMAGE_NAME=$local_image_name MODE=nightly DEVICE=gpu