Damien Benveniste commited on
Commit
6eb29f4
·
1 Parent(s): 0c9fd7a
Files changed (1) hide show
  1. Dockerfile +6 -2
Dockerfile CHANGED
@@ -1,8 +1,12 @@
1
  # Use the official vLLM image as the base image
2
  FROM vllm/vllm-openai:latest
3
 
4
- # If using v0.4.1 or v0.4.2, uncomment the following line:
5
- # ENV VLLM_NCCL_SO_PATH=/root/.config/vllm/nccl/cu12/libnccl.so.2.18.1
 
 
 
 
6
 
7
  # Set the working directory
8
  WORKDIR /app
 
1
  # Use the official vLLM image as the base image
2
  FROM vllm/vllm-openai:latest
3
 
4
+ ENV TRANSFORMERS_CACHE="/tmp/huggingface_cache"
5
+ ENV XDG_CACHE_HOME="/tmp/cache"
6
+
7
+ # Create cache directories and set permissions
8
+ RUN mkdir -p /tmp/huggingface_cache /tmp/cache && \
9
+ chmod 777 /tmp/huggingface_cache /tmp/cache
10
 
11
  # Set the working directory
12
  WORKDIR /app