NIKKI77 commited on
Commit
fe49550
·
1 Parent(s): 215646f

Fix: writable HF caches + OMP threads + libgomp

Browse files
Files changed (1) hide show
  1. Dockerfile +12 -0
Dockerfile CHANGED
@@ -4,6 +4,18 @@ ENV DEBIAN_FRONTEND=noninteractive PIP_NO_CACHE_DIR=1
4
  WORKDIR /app
5
  ENV PYTHONPATH=/app/backend:$PYTHONPATH
6
 
 
 
 
 
 
 
 
 
 
 
 
 
7
  COPY . .
8
 
9
  # Install GPU-ready deps
 
4
  WORKDIR /app
5
  ENV PYTHONPATH=/app/backend:$PYTHONPATH
6
 
7
+ # OpenMP runtime to avoid missing libgomp issues
8
+ RUN apt-get update && apt-get install -y libgomp1 && rm -rf /var/lib/apt/lists/*
9
+
10
+ # Writable caches + sane OMP threads
11
+ ENV HF_HOME=/app/.cache \
12
+ TRANSFORMERS_CACHE=/app/.cache/transformers \
13
+ HUGGINGFACE_HUB_CACHE=/app/.cache/huggingface \
14
+ SENTENCE_TRANSFORMERS_HOME=/app/.cache/sentence-transformers \
15
+ XDG_CACHE_HOME=/app/.cache \
16
+ OMP_NUM_THREADS=1
17
+ RUN mkdir -p /app/.cache && chmod -R 777 /app/.cache
18
+
19
  COPY . .
20
 
21
  # Install GPU-ready deps