adrienbrdne commited on
Commit
f50bdfd
·
verified ·
1 Parent(s): f7bea85

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +5 -0
Dockerfile CHANGED
@@ -9,6 +9,11 @@ ENV PYTHONDONTWRITEBYTECODE 1
9
  # Ensure Python output is sent straight to terminal (useful for logs)
10
  ENV PYTHONUNBUFFERED 1
11
 
 
 
 
 
 
12
  # Upgrade pip
13
  RUN python -m pip install --upgrade pip
14
 
 
9
  # Ensure Python output is sent straight to terminal (useful for logs)
10
  ENV PYTHONUNBUFFERED 1
11
 
12
+
13
+ # Set environment variables for the cache
14
+ ENV HF_HOME="/app/cache"
15
+ ENV TORCH_HOME="/app/cache"
16
+
17
  # Upgrade pip
18
  RUN python -m pip install --upgrade pip
19