diff --git a/.github/workflows/build.yaml b/.github/workflows/build.yaml index 59411588..c1b363c5 100644 --- a/.github/workflows/build.yaml +++ b/.github/workflows/build.yaml @@ -91,9 +91,9 @@ jobs: uses: docker/build-push-action@v4 with: context: . - build-args: - - KERNEL_BUILDER_IMAGE=registry.internal.huggingface.tech/pytorch-base-images/kernel-builder:2.0.0-cuda11.8 - - PYTORCH_IMAGE=registry.internal.huggingface.tech/pytorch-base-images/torch:2.0.0-cuda11.8 + build-args: | + KERNEL_BUILDER_IMAGE=registry.internal.huggingface.tech/pytorch-base-images/kernel-builder:2.0.0-cuda11.8 + PYTORCH_IMAGE=registry.internal.huggingface.tech/pytorch-base-images/torch:2.0.0-cuda11.8 file: Dockerfile push: ${{ github.event_name != 'pull_request' }} platforms: 'linux/amd64'