From 4ac84b8e03db3c2bc30a7a8a592ab40a27bd64e9 Mon Sep 17 00:00:00 2001 From: Param Bole Date: Thu, 16 Jan 2025 23:53:13 +0000 Subject: [PATCH] Adding back builidng images with mode=stable --- .github/workflows/UploadDockerImages.yml | 1 + 1 file changed, 1 insertion(+) diff --git a/.github/workflows/UploadDockerImages.yml b/.github/workflows/UploadDockerImages.yml index 5693b7cb0..33f9a3340 100644 --- a/.github/workflows/UploadDockerImages.yml +++ b/.github/workflows/UploadDockerImages.yml @@ -60,6 +60,7 @@ jobs: run: docker system prune --all --force - name: build jax stable image run : | + bash .github/workflows/build_and_upload_images.sh CLOUD_IMAGE_NAME=maxtext_gpu_jax_stable MODE=stable DEVICE=gpu PROJECT=tpu-prod-env-multipod LOCAL_IMAGE_NAME=maxtext_gpu_local_jax_stable - name: build jax pinned image run : | bash .github/workflows/build_and_upload_images.sh CLOUD_IMAGE_NAME=maxtext_gpu_jax_pinned MODE=pinned DEVICE=gpu PROJECT=tpu-prod-env-multipod LOCAL_IMAGE_NAME=maxtext_gpu_local_jax_pinned