github-actions[bot] commited on
Commit
b966f5f
·
1 Parent(s): 7492a0a

Deploy 2e0cd53

Browse files
Files changed (1) hide show
  1. Dockerfile +6 -1
Dockerfile CHANGED
@@ -3,7 +3,10 @@ FROM python:3.11-slim
3
  ENV PYTHONUNBUFFERED=1 \
4
  PYTHONDONTWRITEBYTECODE=1 \
5
  PIP_NO_CACHE_DIR=1 \
6
- PORT=7860
 
 
 
7
 
8
  WORKDIR /workspace
9
 
@@ -14,6 +17,8 @@ RUN apt-get update && \
14
  libgomp1 \
15
  && rm -rf /var/lib/apt/lists/*
16
 
 
 
17
  COPY . .
18
 
19
  WORKDIR /workspace/apps/api
 
3
  ENV PYTHONUNBUFFERED=1 \
4
  PYTHONDONTWRITEBYTECODE=1 \
5
  PIP_NO_CACHE_DIR=1 \
6
+ PORT=7860 \
7
+ HF_HOME=/tmp/hf-cache \
8
+ TRANSFORMERS_CACHE=/tmp/hf-cache \
9
+ TORCH_HOME=/tmp/torch-cache
10
 
11
  WORKDIR /workspace
12
 
 
17
  libgomp1 \
18
  && rm -rf /var/lib/apt/lists/*
19
 
20
+ RUN mkdir -p "${HF_HOME}" "${TORCH_HOME}" && chmod -R 777 "${HF_HOME}" "${TORCH_HOME}"
21
+
22
  COPY . .
23
 
24
  WORKDIR /workspace/apps/api