Rayugacodes commited on
Commit
74820e1
·
verified ·
1 Parent(s): 3b0341d

Fix: set HF_HOME to writable directory

Browse files
Files changed (1) hide show
  1. Dockerfile +4 -0
Dockerfile CHANGED
@@ -2,6 +2,10 @@ FROM python:3.10-slim
2
 
3
  WORKDIR /app
4
 
 
 
 
 
5
  RUN pip install --no-cache-dir \
6
  torch \
7
  transformers \
 
2
 
3
  WORKDIR /app
4
 
5
+ ENV HF_HOME=/app/.cache/huggingface
6
+ ENV TRANSFORMERS_CACHE=/app/.cache/huggingface
7
+ RUN mkdir -p /app/.cache/huggingface
8
+
9
  RUN pip install --no-cache-dir \
10
  torch \
11
  transformers \