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 c11765ce..b1830cd1 100644 --- a/Dockerfile_trtllm +++ b/Dockerfile_trtllm @@ -80,9 +80,9 @@ ENV SCCACHE_S3_KEY_PREFIX ${SCCACHE_S3_KEY_PREFIX} COPY . . 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