Fix registry name (#2307)

huggingface · Jul 25, 2024 · 3905f85 · 3905f85
1 parent 17ed42b
commit 3905f85
Showing 1 changed file with 5 additions and 6 deletions.
diff --git a/.github/workflows/build.yaml b/.github/workflows/build.yaml
@@ -49,7 +49,7 @@ jobs:
                 export dockerfile="Dockerfile"
                 export label_extension=""
                 export docker_devices=""
-                export runs_on="aws-g6-12xlarge-plus"
+                export runs_on="aws-g6-12xlarge-plus-priv"
                 ;;
             rocm)
                 export dockerfile="Dockerfile_amd"
@@ -75,13 +75,14 @@ jobs:
           echo "LABEL=${label_extension}" >> $GITHUB_ENV
           echo "DOCKER_DEVICES=${docker_devices}" >> $GITHUB_ENV
           echo "RUNS_ON=${runs_on}" >> $GITHUB_ENV
+          echo REGISTRY_MIRROR=$REGISTRY_MIRROR >> $GITHUB_ENV
       - name: Initialize Docker Buildx
         uses: docker/setup-buildx-action@v3
         with:
           install: true
           buildkitd-config-inline: |
             [registry."docker.io"]
-              mirrors = ["registry-us-east-1-mirror.prod.aws.ci.huggingface.tech"]
+              mirrors = ["${{ env.REGISTRY_MIRROR }}"]
       - name: Login to internal Container Registry
         uses: docker/login-action@v3
         with:
@@ -109,7 +110,6 @@ jobs:
         uses: docker/metadata-action@v5
         with:
           images: |
-            registry-us-east-1.prod.aws.ci.huggingface.tech/api-inference/community/text-generation-inference
             registry.internal.huggingface.tech/api-inference/community/text-generation-inference
           tags: |
             type=raw,value=sha-${{ env.GITHUB_SHA_SHORT }}${{ env.LABEL }}
@@ -122,8 +122,7 @@ jobs:
           flavor: |
             latest=auto
           images: |
-            registry-us-east-1.prod.aws.ci.huggingface.tech/api-inference/community/text-generation-inference
-            registry.internal.huggingface.tech/api-inference/community/text-generation-inferenceca
+            registry.internal.huggingface.tech/api-inference/community/text-generation-inference
             ghcr.io/huggingface/text-generation-inference
             db4c2190dd824d1f950f5d1555fbadf0.azurecr.io/text-generation-inference
           tags: |
@@ -149,7 +148,7 @@ jobs:
       - name: Final
         id: final
         run: |
-          echo "docker_image=registry-us-east-1.prod.aws.ci.huggingface.tech/api-inference/community/text-generation-inference:sha-${{ env.GITHUB_SHA_SHORT}}${{ env.LABEL }}" >> "$GITHUB_OUTPUT"
+          echo "docker_image=registry.internal.huggingface.tech/api-inference/community/text-generation-inference:sha-${{ env.GITHUB_SHA_SHORT}}${{ env.LABEL }}" >> "$GITHUB_OUTPUT"
           echo "docker_devices=${{ env.DOCKER_DEVICES }}" >> "$GITHUB_OUTPUT"
           echo "runs_on=${{ env.RUNS_ON }}" >> "$GITHUB_OUTPUT"
           echo "label=${{ env.LABEL }}" >> "$GITHUB_OUTPUT"