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
......@@ -191,8 +191,10 @@ ARG HAS_TRTLLM_CONTEXT
ARG TENSORRTLLM_PIP_WHEEL
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 /*.txt /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
# 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