Gunroar commited on
Commit
948851b
·
1 Parent(s): a7c894d

add changes

Browse files
Files changed (1) hide show
  1. Dockerfile +6 -2
Dockerfile CHANGED
@@ -5,8 +5,9 @@ FROM python:3.10-slim
5
  ENV PYTHONDONTWRITEBYTECODE=1 \
6
  PYTHONUNBUFFERED=1 \
7
  PIP_NO_CACHE_DIR=1 \
8
- HF_HOME=/root/.cache/huggingface \
9
- TRANSFORMERS_CACHE=/root/.cache/huggingface/transformers \
 
10
  TOKENIZERS_PARALLELISM=false
11
 
12
  WORKDIR /app
@@ -17,6 +18,9 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
17
  git \
18
  && rm -rf /var/lib/apt/lists/*
19
 
 
 
 
20
  # Copy only requirements first for better layer caching
21
  COPY requirements.txt ./
22
 
 
5
  ENV PYTHONDONTWRITEBYTECODE=1 \
6
  PYTHONUNBUFFERED=1 \
7
  PIP_NO_CACHE_DIR=1 \
8
+ HF_HOME=/app/hf-home \
9
+ HUGGINGFACE_HUB_CACHE=/app/hf-home \
10
+ TRANSFORMERS_CACHE=/app/hf-home/transformers \
11
  TOKENIZERS_PARALLELISM=false
12
 
13
  WORKDIR /app
 
18
  git \
19
  && rm -rf /var/lib/apt/lists/*
20
 
21
+ # Ensure Hugging Face cache dirs exist and are writable
22
+ RUN mkdir -p /app/hf-home/transformers && chmod -R 777 /app/hf-home
23
+
24
  # Copy only requirements first for better layer caching
25
  COPY requirements.txt ./
26