From 30617dfd442727d221b633a7f6b447941dbe7f44 Mon Sep 17 00:00:00 2001 From: makaveli10 Date: Tue, 3 Dec 2024 18:27:11 +0530 Subject: [PATCH] Checkout git tensorrt_llm==v0.15.0 in dockerfile Signed-off-by: makaveli10 --- docker/Dockerfile.tensorrt | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docker/Dockerfile.tensorrt b/docker/Dockerfile.tensorrt index 306b015..742e494 100644 --- a/docker/Dockerfile.tensorrt +++ b/docker/Dockerfile.tensorrt @@ -9,7 +9,7 @@ RUN apt-get update && apt-get install -y \ FROM base AS devel RUN pip3 install --no-cache-dir -U tensorrt_llm==0.15.0.dev2024111200 --extra-index-url https://pypi.nvidia.com WORKDIR /app -RUN git clone --depth 1 https://github.com/NVIDIA/TensorRT-LLM.git && cd TensorRT-LLM && \ +RUN git clone https://github.com/NVIDIA/TensorRT-LLM.git && cd TensorRT-LLM && \ git checkout c629546ce429623c8a163633095230154a6f0574 && cd ../ && \ mv TensorRT-LLM/examples ./TensorRT-LLM-examples && \ rm -rf TensorRT-LLM