Skip to content

GitLab

Unverified Commit 3d7f7a56 authored by Anant Sharma's avatar Anant Sharma Committed by GitHub
Browse files

fix: copy commit info in trtllm build (#3619) 


Signed-off-by: default avatarAnant Sharma <anants@nvidia.com>
parent 9a002b30
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 1 deletion
container/Dockerfile.trtllm
View file @ 3d7f7a56
...@@ -191,8 +191,10 @@ ARG HAS_TRTLLM_CONTEXT ...@@ -191,8 +191,10 @@ ARG HAS_TRTLLM_CONTEXT
ARG TENSORRTLLM_PIP_WHEEL ARG TENSORRTLLM_PIP_WHEEL
ARG TENSORRTLLM_INDEX_URL ARG TENSORRTLLM_INDEX_URL
# Copy only wheel files from trtllm_wheel stage from build_context # Copy only wheel files and commit info from trtllm_wheel stage from build_context
COPY --from=trtllm_wheel /*.whl /trtllm_wheel/ COPY --from=trtllm_wheel /*.whl /trtllm_wheel/
COPY --from=trtllm_wheel /*.txt /trtllm_wheel/
# Note: TensorRT needs to be uninstalled before installing the TRTLLM wheel # Note: TensorRT needs to be uninstalled before installing the TRTLLM wheel
# because there might be mismatched versions of TensorRT between the NGC PyTorch # because there might be mismatched versions of TensorRT between the NGC PyTorch
# and the TRTLLM wheel. # and the TRTLLM wheel.
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment