limcheekin commited on
Commit
e1911b7
1 Parent(s): bcd9fab

chore: added TRANSFORMERS_CACHE environment variable

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -0
Dockerfile CHANGED
@@ -25,6 +25,7 @@ FROM python:3.10-slim
25
  ARG MODEL
26
  ENV MODEL=${MODEL}
27
  ENV NORMALIZE_EMBEDDINGS=1
 
28
  # Set environment variable for the host
29
  ENV HOST=0.0.0.0
30
  ENV PORT=7860
 
25
  ARG MODEL
26
  ENV MODEL=${MODEL}
27
  ENV NORMALIZE_EMBEDDINGS=1
28
+ ENV TRANSFORMERS_CACHE="/tmp/transformers_cache"
29
  # Set environment variable for the host
30
  ENV HOST=0.0.0.0
31
  ENV PORT=7860