diff --git a/CHANGELOG.rst b/CHANGELOG.rst index a52771f2..fffda483 100755 --- a/CHANGELOG.rst +++ b/CHANGELOG.rst @@ -23,6 +23,7 @@ Model Optimizer Changelog (Linux) **Bug Fixes** - Fix attention head ranking logic for pruning Megatron Core GPT models. +- Upgrade TensorRT-LLM dependency to 1.1.0rc2. **New Features** diff --git a/docker/Dockerfile b/docker/Dockerfile index 7776af71..3fd96a09 100644 --- a/docker/Dockerfile +++ b/docker/Dockerfile @@ -1,4 +1,4 @@ -FROM nvcr.io/nvidia/tensorrt-llm/release:1.0.0rc6 +FROM nvcr.io/nvidia/tensorrt-llm/release:1.1.0rc2.post2 ARG PIP_EXTRA_INDEX_URL="https://pypi.nvidia.com" ENV PIP_EXTRA_INDEX_URL=$PIP_EXTRA_INDEX_URL \