Skip to content

Commit

Permalink
Updated Branch
Browse files Browse the repository at this point in the history
  • Loading branch information
parambole committed Jul 22, 2024
1 parent 426ef52 commit ade3ec6
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions .github/workflows/UploadDockerImages.yml
Original file line number Diff line number Diff line change
Expand Up @@ -19,7 +19,7 @@ name: Build Images

on:
push:
branches: [ "main" ]
branches: [ "pb-jax-ss" ]

jobs:
tpu:
Expand All @@ -32,4 +32,4 @@ jobs:
- uses: actions/checkout@v3
- name: build jax stable stack image
run : |
bash docker_maxtext_jax_ss_image_upload.sh PROJECT_ID=tpu-prod-env-multipod BASEIMAGE=us-docker.pkg.dev/tpu-prod-env-multipod/jax-ss/tpu:jax0.4.30-rev1 CLOUD_IMAGE_NAME=maxtext-jax-ss IMAGE_TAG=jax0.4.30-rev1 MAXTEXT_REQUIREMENTS_FILE=requirements_with_jax_ss.txt
bash docker_maxdiffusion_image_upload.sh PROJECT_ID=tpu-prod-env-multipod BASEIMAGE=us-docker.pkg.dev/tpu-prod-env-multipod/jax-ss/tpu:jax0.4.30-rev1 CLOUD_IMAGE_NAME=maxdiffusion-jax-ss IMAGE_TAG=jax0.4.30-rev1 MAXDIFFUISON_REQUIREMENTS_FILE=requirements_with_jax_ss.txt

0 comments on commit ade3ec6

Please sign in to comment.