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

chore: fix dev containers

Browse Source
This commit is contained in:
Hugo Larcher 2024-12-02 16:42:42 +01:00
parent eb6698f1ed
commit 5acb1a6390
No known key found for this signature in database
GPG Key ID: 3DAF63124699CA2B
2 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
.devcontainer/Dockerfile.trtllm → .devcontainer/Dockerfile_trtllm
View File

@ -71,3 +71,5 @@ RUN cargo install cargo-chef
COPY --from=trt-builder /usr/local/tensorrt /usr/local/tensorrt COPY --from=trt-builder /usr/local/tensorrt /usr/local/tensorrt
COPY --from=mpi-builder /usr/local/mpi /usr/local/mpi COPY --from=mpi-builder /usr/local/mpi /usr/local/mpi
ENV MPI_HOME=/usr/local/mpi

2
.devcontainer/devcontainer.json
View File

@ -4,7 +4,7 @@
"name": "CUDA", "name": "CUDA",
// "image": "nvidia/cuda:12.6.0-devel-ubuntu24.04", // "image": "nvidia/cuda:12.6.0-devel-ubuntu24.04",
"build": { "build": {
"dockerfile": "Dockerfile.trtllm" "dockerfile": "Dockerfile_trtllm"
}, },
"remoteEnv": { "remoteEnv": {
"PATH": "${containerEnv:PATH}:/usr/local/cuda/bin", "PATH": "${containerEnv:PATH}:/usr/local/cuda/bin",