krislette commited on
Commit
8074d06
·
1 Parent(s): 769cf16

Restored transformers cache

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -0
Dockerfile CHANGED
@@ -45,6 +45,7 @@ RUN mkdir -p /app/.cache/huggingface /app/.cache/torch /tmp/numba_cache && \
45
  # Set environment
46
  ENV PYTHONPATH="/app"
47
  ENV HF_HOME="/app/.cache/huggingface"
 
48
  ENV TORCH_HOME="/app/.cache/torch"
49
  ENV NUMBA_CACHE_DIR="/tmp/numba_cache"
50
  ENV NUMBA_DISABLE_JIT=0
 
45
  # Set environment
46
  ENV PYTHONPATH="/app"
47
  ENV HF_HOME="/app/.cache/huggingface"
48
+ ENV TRANSFORMERS_CACHE="/app/.cache/huggingface"
49
  ENV TORCH_HOME="/app/.cache/torch"
50
  ENV NUMBA_CACHE_DIR="/tmp/numba_cache"
51
  ENV NUMBA_DISABLE_JIT=0