TensorRT-LLM provides users with an easy-to-use Python API to define Large Language Models (LLMs) and build TensorRT engines that contain state-of-the-art optimizations to perform inference efficiently on NVIDIA GPUs. TensorRT-LLM also contains components to create Python and C++ runtimes that execute those TensorRT engines.
--weight_only_precision
argument fromtrtllm-build
command.shared_embedding_table
is not being set when loading Gemma #1799, thanks to the contribution from @mfuntowicz.gptManagerBenchmark
.nvcr.io/nvidia/pytorch:24.05-py3
.nvcr.io/nvidia/tritonserver:24.05-py3
.