Skip to content

Commit 8d0e40f

Browse files
authored
Upgrade TensorRT-LLM docker to 1.1.0RC2 (#327)
Signed-off-by: Chenjie Luo <[email protected]>
1 parent 8a5736a commit 8d0e40f

File tree

2 files changed

+2
-1
lines changed

2 files changed

+2
-1
lines changed

CHANGELOG.rst

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -23,6 +23,7 @@ Model Optimizer Changelog (Linux)
2323
**Bug Fixes**
2424

2525
- Fix attention head ranking logic for pruning Megatron Core GPT models.
26+
- Upgrade TensorRT-LLM dependency to 1.1.0rc2.
2627

2728
**New Features**
2829

docker/Dockerfile

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
FROM nvcr.io/nvidia/tensorrt-llm/release:1.0.0rc6
1+
FROM nvcr.io/nvidia/tensorrt-llm/release:1.1.0rc2.post2
22

33
ARG PIP_EXTRA_INDEX_URL="https://pypi.nvidia.com"
44
ENV PIP_EXTRA_INDEX_URL=$PIP_EXTRA_INDEX_URL \

0 commit comments

Comments
 (0)