From d72fc1014724ffd17955d1c5503cce7c108fad1b Mon Sep 17 00:00:00 2001 From: Krzysztof G Date: Sun, 2 Jun 2024 15:09:11 +0200 Subject: [PATCH] Update TensorRT repository branch to match TensorRT installed --- Popular_Models_Guide/StableDiffusion/docker/Dockerfile | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/Popular_Models_Guide/StableDiffusion/docker/Dockerfile b/Popular_Models_Guide/StableDiffusion/docker/Dockerfile index 6f7c050e..1e351f22 100644 --- a/Popular_Models_Guide/StableDiffusion/docker/Dockerfile +++ b/Popular_Models_Guide/StableDiffusion/docker/Dockerfile @@ -31,7 +31,7 @@ FROM ${BASE_IMAGE}:${BASE_IMAGE_TAG} as tritonserver-stable-diffusion RUN pip install --pre --upgrade --extra-index-url https://pypi.nvidia.com tensorrt -RUN git clone https://github.com/NVIDIA/TensorRT.git -b release/9.2 --single-branch /tmp/TensorRT +RUN git clone https://github.com/NVIDIA/TensorRT.git -b release/10.0 --single-branch /tmp/TensorRT RUN pip3 install -r /tmp/TensorRT/demo/Diffusion/requirements.txt