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