configure pkg.dev

stanford-crfm · Sep 6, 2024 · 277e5b8 · 277e5b8
1 parent 314bd99
commit 277e5b8
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/.github/workflows/launch_small_fast.yaml b/.github/workflows/launch_small_fast.yaml
@@ -34,16 +34,15 @@ jobs:
       - name: Configure Google Cloud
         run: |
           gcloud config set project ${{ secrets.GCP_PROJECT_ID }}
+          REGION=${TPU_ZONE%-*}
+          echo "$REGION"
+          gcloud auth configure-docker  $REGION-docker.pkg.dev
 
       - name: Install locally
         run: |
           python -m pip install --upgrade pip
           pip install -e .[test] "jax[cpu]==0.4.30"
 
-      - name: Do we have docker?
-        run: |
-          docker --version
-
       - name: Launch Small Fast TPU Train LM job
         run: |
           export TPU_NAME=small-fast-${{ github.run_id }}