From dc34f5a70edca2156e85795c60e0025aa2cbfb1b Mon Sep 17 00:00:00 2001 From: Morgan Funtowicz Date: Thu, 12 Dec 2024 14:24:33 +0100 Subject: [PATCH] misc(ci): again --- .github/workflows/build_trtllm.yaml | 1 + Dockerfile_trtllm | 6 +++--- 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/.github/workflows/build_trtllm.yaml b/.github/workflows/build_trtllm.yaml index 9d62558b..1f67bca0 100644 --- a/.github/workflows/build_trtllm.yaml +++ b/.github/workflows/build_trtllm.yaml @@ -52,6 +52,7 @@ jobs: file: Dockerfile_trtllm target: runtime push: false + load: true platforms: 'linux/amd64' build-args: | BUILD_TYPE=debug diff --git a/Dockerfile_trtllm b/Dockerfile_trtllm index 68428219..e61dfaec 100644 --- a/Dockerfile_trtllm +++ b/Dockerfile_trtllm @@ -95,9 +95,9 @@ COPY router router COPY backends/trtllm backends/trtllm COPY --from=trt-builder /usr/local/tensorrt /usr/local/tensorrt COPY --from=mpi-builder /usr/local/mpi /usr/local/mpi -RUN --mount=type=secret,id=AWS_ACCESS_KEY_ID \ - --mount=type=secret,id=AWS_SECRET_ACCESS_KEY \ - --mount=type=secret,id=SCCACHE_BUCKET \ +RUN --mount=type=secret,id=AWS_ACCESS_KEY_ID,env=AWS_ACCESS_KEY_ID \ + --mount=type=secret,id=AWS_SECRET_ACCESS_KEY,env=AWS_SECRET_ACCESS_KEY \ + --mount=type=secret,id=SCCACHE_BUCKET,env=SCCACHE_BUCKET \ mkdir $TGI_INSTALL_PREFIX && mkdir "$TGI_INSTALL_PREFIX/include" && mkdir "$TGI_INSTALL_PREFIX/lib" && \ CMAKE_INSTALL_PREFIX=$TGI_INSTALL_PREFIX cargo build --${BUILD_TYPE} --package text-generation-backends-trtllm --bin text-generation-backends-trtllm