Steveeeeeeen HF staff commited on
Commit
b383ad5
·
verified ·
1 Parent(s): d743fc1

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -1
Dockerfile CHANGED
@@ -47,7 +47,7 @@ RUN git clone https://github.com/Zyphra/Zonos.git && cd Zonos
47
 
48
  # Set environment variables for writable cache directories
49
  ENV TRITON_CACHE_DIR=/tmp/.triton
50
- ENV TRANSFORMERS_CACHE=/tmp/huggingface_cache
51
 
52
  # Ensure cache directories are writable
53
  RUN mkdir -p $TRITON_CACHE_DIR $TRANSFORMERS_CACHE && chmod -R 777 $TRITON_CACHE_DIR $TRANSFORMERS_CACHE
 
47
 
48
  # Set environment variables for writable cache directories
49
  ENV TRITON_CACHE_DIR=/tmp/.triton
50
+ ENV HF_HOME=/tmp/huggingface_cache
51
 
52
  # Ensure cache directories are writable
53
  RUN mkdir -p $TRITON_CACHE_DIR $TRANSFORMERS_CACHE && chmod -R 777 $TRITON_CACHE_DIR $TRANSFORMERS_CACHE