huggingface/text-generation-inference
5
0
Fork 0
mirror of https://github.com/huggingface/text-generation-inference.git synced 2025-09-11 12:24:53 +00:00
Code Issues Packages Projects Releases Wiki Activity

backend(trtllm): inject ompi_version build arg in dependent step

Browse Source
This commit is contained in:
Morgan Funtowicz 2025-01-26 15:21:38 +01:00
parent c632f8a95a
commit 5a317ffad7
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
Dockerfile_trtllm
View File

@ -38,6 +38,7 @@ ENV TENSORRT_INSTALL_PREFIX=/usr/local/tensorrt
FROM cuda-builder AS mpi-builder FROM cuda-builder AS mpi-builder
WORKDIR /opt/src/mpi WORKDIR /opt/src/mpi
ARG ompi_version
ENV OMPI_VERSION=${ompi_version} ENV OMPI_VERSION=${ompi_version}
ENV OMPI_TARBALL_FILENAME=openmpi-${OMPI_VERSION}.tar.bz2 ENV OMPI_TARBALL_FILENAME=openmpi-${OMPI_VERSION}.tar.bz2
ADD --checksum=sha256:54a33cb7ad81ff0976f15a6cc8003c3922f0f3d8ceed14e1813ef3603f22cd34 \ ADD --checksum=sha256:54a33cb7ad81ff0976f15a6cc8003c3922f0f3d8ceed14e1813ef3603f22cd34 \