From bd0f7540ff013581c242ba6fbda89253255b52bf Mon Sep 17 00:00:00 2001 From: Rohan Bierneni Date: Fri, 15 Aug 2025 17:39:33 +0000 Subject: [PATCH] Add nightly JAX AI Image build to maxdiffusion --- .github/workflows/UploadDockerImages.yml | 3 +++ 1 file changed, 3 insertions(+) diff --git a/.github/workflows/UploadDockerImages.yml b/.github/workflows/UploadDockerImages.yml index 4af852141..9231e556a 100644 --- a/.github/workflows/UploadDockerImages.yml +++ b/.github/workflows/UploadDockerImages.yml @@ -32,6 +32,9 @@ jobs: - name: build maxdiffusion jax ai image run: | bash .github/workflows/build_and_upload_images.sh CLOUD_IMAGE_NAME=maxdiffusion_jax_stable_stack MODE=jax_ai_image PROJECT=tpu-prod-env-multipod LOCAL_IMAGE_NAME=maxdiffusion_jax_stable_stack BASEIMAGE=us-docker.pkg.dev/cloud-tpu-images/jax-ai-image/tpu:latest + - name: build maxdiffusion w/ nightly jax ai image + run: | + bash .github/workflows/build_and_upload_images.sh CLOUD_IMAGE_NAME=maxdiffusion_jax_stable_stack_nightly MODE=jax_ai_image PROJECT=tpu-prod-env-multipod LOCAL_IMAGE_NAME=maxdiffusion_jax_stable_stack BASEIMAGE=us-docker.pkg.dev/tpu-prod-env-multipod/jax-stable-stack/tpu/jax_nightly:latest - name: build maxdiffusion jax nightly image run: | bash .github/workflows/build_and_upload_images.sh CLOUD_IMAGE_NAME=maxdiffusion_jax_nightly MODE=nightly PROJECT=tpu-prod-env-multipod LOCAL_IMAGE_NAME=maxdiffusion_jax_nightly