yusufs commited on
Commit
f918ab8
·
verified ·
1 Parent(s): 2d28796

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +8 -0
Dockerfile CHANGED
@@ -1,6 +1,14 @@
1
 
2
  FROM vllm/vllm-openai:v0.10.0
3
 
 
 
 
 
 
 
 
 
4
  # e.g. install the `audio` optional dependencies
5
  # NOTE: Make sure the version of vLLM matches the base image!
6
  RUN uv pip install --system vllm[audio]==0.10.0
 
1
 
2
  FROM vllm/vllm-openai:v0.10.0
3
 
4
+ ENV VLLM_LOGGING_LEVEL=DEBUG
5
+ ENV HF_HOME=/tmp/.cache/huggingface
6
+ ENV OMP_NUM_THREADS=1
7
+ ENV VLLM_USE_TRITON_FLASH_ATTN=0
8
+ ENV VLLM_ATTENTION_BACKEND=XFORMERS
9
+
10
+ RUN mkdir -p /tmp/.cache/huggingface
11
+
12
  # e.g. install the `audio` optional dependencies
13
  # NOTE: Make sure the version of vLLM matches the base image!
14
  RUN uv pip install --system vllm[audio]==0.10.0