jobian commited on
Commit
7b412b0
·
1 Parent(s): d8475f5
Files changed (1) hide show
  1. Dockerfile +1 -1
Dockerfile CHANGED
@@ -8,7 +8,7 @@ WORKDIR /app
8
  COPY . .
9
 
10
  # Set cache environment so model is stored in /app/.cache
11
- ENV TRANSFORMERS_CACHE=/app/.cache/transformers
12
  ENV HF_HUB_CACHE=/app/.cache/hub
13
 
14
  # Pre-download the model
 
8
  COPY . .
9
 
10
  # Set cache environment so model is stored in /app/.cache
11
+ ENV HF_HOME=/app/.cache/transformers
12
  ENV HF_HUB_CACHE=/app/.cache/hub
13
 
14
  # Pre-download the model