diff --git a/.github/workflows/build.yaml b/.github/workflows/build.yaml index c289f63f..5165ae1f 100644 --- 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"