binary1ne commited on
Commit
d05305c
·
verified ·
1 Parent(s): 306df80

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +7 -1
Dockerfile CHANGED
@@ -2,8 +2,14 @@ FROM harshmanvar/vllm-cpu-only:v1
2
 
3
  # Set writable Hugging Face cache directory
4
  ENV TRANSFORMERS_CACHE=/workspace/hf_cache
 
5
  ENV HF_HOME=/workspace/hf_cache
6
- RUN mkdir -p /workspace/hf_cache && chmod -R 777 /workspace/hf_cache
 
 
 
 
 
7
 
8
  RUN pip show vllm
9
 
 
2
 
3
  # Set writable Hugging Face cache directory
4
  ENV TRANSFORMERS_CACHE=/workspace/hf_cache
5
+ # Set Hugging Face cache dir
6
  ENV HF_HOME=/workspace/hf_cache
7
+ RUN mkdir -p $HF_HOME && chmod -R 777 $HF_HOME
8
+
9
+ # Install git & git-lfs
10
+ RUN apt-get update && apt-get install -y git git-lfs && \
11
+ git lfs install && \
12
+ apt-get clean && rm -rf /var/lib/apt/lists/*
13
 
14
  RUN pip show vllm
15