enskaff commited on
Commit
84c5bef
·
verified ·
1 Parent(s): 5a60ff4

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +3 -1
Dockerfile CHANGED
@@ -8,7 +8,9 @@ ENV DEBIAN_FRONTEND=noninteractive \
8
  PYTHONIOENCODING=utf-8 \
9
  PYTHONUNBUFFERED=1 \
10
  HF_HOME=/app/huggingface_cache \
11
- NUMBA_DISABLE_CACHE=1
 
 
12
 
13
  # Create the transformers cache directory and ensure it's writable
14
  RUN mkdir -p /app/transformers_cache && chmod -R 777 /app/transformers_cache
 
8
  PYTHONIOENCODING=utf-8 \
9
  PYTHONUNBUFFERED=1 \
10
  HF_HOME=/app/huggingface_cache \
11
+ TRANSFORMERS_CACHE=/app/huggingface_cache \
12
+ NUMBA_DISABLE_CACHE=1 \
13
+ NUMBA_CACHE_DIR=/tmp/numba_cache
14
 
15
  # Create the transformers cache directory and ensure it's writable
16
  RUN mkdir -p /app/transformers_cache && chmod -R 777 /app/transformers_cache