Update Dockerfile
Browse files- 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 |
|