limcheekin commited on
Commit
e4dee70
1 Parent(s): db8aa1c

fix: changed TRANSFORMERS_CACHE to HF_HOME

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -1
Dockerfile CHANGED
@@ -24,7 +24,7 @@ FROM python:3.11-slim
24
  # Include global args in this stage of the build
25
  ARG MODEL
26
  ENV MODEL=${MODEL}
27
- ENV TRANSFORMERS_CACHE="/tmp/transformers_cache"
28
  # Set environment variable for the host
29
  ENV HOST=0.0.0.0
30
  ENV PORT=7860
 
24
  # Include global args in this stage of the build
25
  ARG MODEL
26
  ENV MODEL=${MODEL}
27
+ ENV HF_HOME="/tmp/hf_home"
28
  # Set environment variable for the host
29
  ENV HOST=0.0.0.0
30
  ENV PORT=7860