Skip to content

Commit a645748

Browse files
authored
Merge pull request #169 from NillionNetwork/chore/update_vllm_0_11
feat: update vLLM version 0.11.2
2 parents 028dc61 + d82751d commit a645748

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

docker/vllm.Dockerfile

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
FROM vllm/vllm-openai:v0.10.1
1+
FROM vllm/vllm-openai:v0.11.2
22

33
# # Specify model name and path during build
44
# ARG MODEL_NAME=llama_1b_cpu
@@ -9,6 +9,7 @@ FROM vllm/vllm-openai:v0.10.1
99
# ENV MODEL_PATH=${MODEL_PATH}
1010
# ENV EXEC_PATH=nilai_models.models.${MODEL_NAME}:app
1111

12+
ENV PYTORCH_CUDA_ALLOC_CONF=expandable_segments:True
1213
COPY --link . /daemon/
1314
COPY --link vllm_templates /opt/vllm/templates
1415

0 commit comments

Comments
 (0)