From 5327674f5d9534d44d14a19d1268a3bb13c7f03a Mon Sep 17 00:00:00 2001 From: Siddharth Venkatesan Date: Wed, 13 Nov 2024 13:21:48 -0800 Subject: [PATCH] [trt][docker] pin trt version to release version, not nightly --- serving/docker/tensorrt-llm.Dockerfile | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/serving/docker/tensorrt-llm.Dockerfile b/serving/docker/tensorrt-llm.Dockerfile index 9e1dba241..5a9b3f4f2 100644 --- a/serving/docker/tensorrt-llm.Dockerfile +++ b/serving/docker/tensorrt-llm.Dockerfile @@ -20,7 +20,7 @@ ARG transformers_version=4.44.2 ARG accelerate_version=0.33.0 ARG tensorrtlibs_version=10.3.0 # %2B is the url escape for the '+' character -ARG trtllm_toolkit_version=0.12.0%2Bnightly +ARG trtllm_toolkit_version=0.12.0 ARG trtllm_version=v0.12.0 ARG cuda_python_version=12.5 ARG peft_version=0.10.0