Bagi4 commited on
Commit
5576075
1 Parent(s): 21a3c73

fix: Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -0
Dockerfile CHANGED
@@ -2,6 +2,7 @@ FROM python:3.9
2
 
3
  COPY . .
4
 
 
5
  RUN pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
6
  RUN pip install --upgrade -r requirements.txt
7
 
 
2
 
3
  COPY . .
4
 
5
+ RUN export TRANSFORMERS_CACHE=/blabla/cache/
6
  RUN pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
7
  RUN pip install --upgrade -r requirements.txt
8