devve1 commited on
Commit
31c273d
1 Parent(s): d644561

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -0
Dockerfile CHANGED
@@ -46,6 +46,7 @@ ENV HOME=/home/user \
46
  NVIDIA_DRIVER_CAPABILITIES=compute,utility \
47
  SYSTEM=spaces \
48
  HF_HOME=$HOME/data \
 
49
  NLTK_DATA=$HF_HOME \
50
  TOKENIZERS_PARALLELISM=false \
51
  PYTORCH_CUDA_ALLOC_CONF=expandable_segments:True
 
46
  NVIDIA_DRIVER_CAPABILITIES=compute,utility \
47
  SYSTEM=spaces \
48
  HF_HOME=$HOME/data \
49
+ OUTLINES_CACHE_DIR=$HOME/data \
50
  NLTK_DATA=$HF_HOME \
51
  TOKENIZERS_PARALLELISM=false \
52
  PYTORCH_CUDA_ALLOC_CONF=expandable_segments:True