From 32e8c06a1a9c4ec2e681ef49dc84dbb17e2e9e45 Mon Sep 17 00:00:00 2001 From: OlivierDehaene <23298448+OlivierDehaene@users.noreply.github.com> Date: Tue, 11 Apr 2023 17:54:36 +0200 Subject: [PATCH] fix build --- .github/workflows/build.yaml | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) 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'