Update Dockerfile
Browse files- Dockerfile +4 -0
Dockerfile
CHANGED
|
@@ -14,6 +14,10 @@ FROM nvidia/cuda:12.1.1-cudnn8-runtime-ubuntu22.04 AS backend
|
|
| 14 |
|
| 15 |
ENV DEBIAN_FRONTEND=noninteractive PYTHONDONTWRITEBYTECODE=1 PYTHONUNBUFFERED=1 PIP_NO_CACHE_DIR=1
|
| 16 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 17 |
# Install Python and system deps
|
| 18 |
RUN apt-get update && apt-get install -y --no-install-recommends python3 python3-pip python3-venv git && rm -rf /var/lib/apt/lists/*
|
| 19 |
|
|
|
|
| 14 |
|
| 15 |
ENV DEBIAN_FRONTEND=noninteractive PYTHONDONTWRITEBYTECODE=1 PYTHONUNBUFFERED=1 PIP_NO_CACHE_DIR=1
|
| 16 |
|
| 17 |
+
ENV HF_HOME=/app/.cache/huggingface
|
| 18 |
+
ENV TRANSFORMERS_CACHE=/app/.cache/huggingface/transformers
|
| 19 |
+
|
| 20 |
+
RUN mkdir -p /app/.cache/huggingface
|
| 21 |
# Install Python and system deps
|
| 22 |
RUN apt-get update && apt-get install -y --no-install-recommends python3 python3-pip python3-venv git && rm -rf /var/lib/apt/lists/*
|
| 23 |
|