Adding back builidng images with mode=stable

AI-Hypercomputer · Jan 16, 2025 · 4ac84b8 · 4ac84b8
1 parent 3ad02ba
commit 4ac84b8
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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