binary1ne commited on
Commit
2ff3a36
·
verified ·
1 Parent(s): bec8dc0

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -0
Dockerfile CHANGED
@@ -2,6 +2,7 @@ FROM harshmanvar/vllm-cpu-only:v1
2
 
3
  # Set writable Hugging Face cache directory
4
  ENV TRANSFORMERS_CACHE=/workspace/hf_cache
 
5
  RUN mkdir -p /workspace/hf_cache && chmod -R 777 /workspace/hf_cache
6
 
7
  RUN pip show vllm
 
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