Bushra-KB commited on
Commit
5b2c761
·
verified ·
1 Parent(s): 0b65204

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -1
Dockerfile CHANGED
@@ -6,7 +6,7 @@ ENV HF_HOME=/data/huggingface
6
  ENV TORCH_HOME=/data/torch
7
  ENV TRANSFORMERS_CACHE=/data/huggingface/hub
8
  ENV HF_DATASETS_CACHE=/data/huggingface/datasets
9
- ENV HOME=/tmp # Set HOME to /tmp for any fallback cache writes
10
 
11
  # Create the cache directories and make them writable by any user
12
  RUN mkdir -p $TRANSFORMERS_CACHE $HF_DATASETS_CACHE $TORCH_HOME && \
 
6
  ENV TORCH_HOME=/data/torch
7
  ENV TRANSFORMERS_CACHE=/data/huggingface/hub
8
  ENV HF_DATASETS_CACHE=/data/huggingface/datasets
9
+ ENV HOME=/tmp
10
 
11
  # Create the cache directories and make them writable by any user
12
  RUN mkdir -p $TRANSFORMERS_CACHE $HF_DATASETS_CACHE $TORCH_HOME && \