new runner

AI-Hypercomputer · Jul 24, 2024 · 77624eb · 77624eb
1 parent e227628
commit 77624eb
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 7 deletions.
diff --git a/.github/workflows/UploadDockerImages.yml b/.github/workflows/UploadDockerImages.yml
@@ -22,11 +22,10 @@ on:
     branches: [ "pb-jax-ss" ]
 
 jobs:
-  Build-Image-On-Ubuntu:
-    name: Build Maxdiffusion JAX Stable Stack Image
-    runs-on: ubuntu-latest
+  build-image:
+    runs-on: ["self-hosted", "maxdiffusion-github-docker-image-build-runner"]
     steps:
-    - uses: actions/checkout@v4
+    - uses: actions/checkout@v3
     - name: build jax stable stack image
       run: |
-        bash docker_maxdiffusion_image_upload.sh PROJECT_ID=tpu-prod-env-multipod BASEIMAGE=gcr.io/tpu-prod-env-multipod/jax-stable-stack/tpu:jax0.4.30-rev1 CLOUD_IMAGE_NAME=maxdiffusion-jax-stable-stack IMAGE_TAG=jax0.4.30-rev1 MAXDIFFUSION_REQUIREMENTS_FILE=requirements_with_jax_stable_stack.txt DELETE_LOCAL_IMAGE=true
+        bash docker_maxdiffusion_image_upload.sh PROJECT_ID=tpu-prod-env-multipod BASEIMAGE=us-docker.pkg.dev/tpu-prod-env-multipod/jax-stable-stack/tpu:jax0.4.30-rev1 CLOUD_IMAGE_NAME=maxdiffusion-jax-stable-stack IMAGE_TAG=jax0.4.30-rev1 MAXDIFFUSION_REQUIREMENTS_FILE=requirements_with_jax_stable_stack.txt DELETE_LOCAL_IMAGE=true
diff --git a/docker_maxdiffusion_image_upload.sh b/docker_maxdiffusion_image_upload.sh
@@ -62,8 +62,6 @@ fi
 # Default: Don't delete local image
 DELETE_LOCAL_IMAGE="${DELETE_LOCAL_IMAGE:-false}"
 
-gcloud auth configure-docker us-docker.pkg.dev --quiet
-
 COMMIT_HASH=$(git rev-parse --short HEAD)
 
 echo "Building JAX Stable Stack MaxDiffusion at commit hash ${COMMIT_HASH} . . ."