jwphantom commited on
Commit
503c42b
1 Parent(s): e892487

fix problem get_relevant_documents

Browse files
Files changed (1) hide show
  1. Dockerfile +4 -0
Dockerfile CHANGED
@@ -12,6 +12,10 @@ RUN pip wheel --no-cache-dir --no-deps --wheel-dir /app/wheels -r requirements.t
12
  # Étape finale
13
  FROM python:3.11
14
 
 
 
 
 
15
  WORKDIR /app
16
 
17
  # Créer le répertoire de cache et définir les permissions
 
12
  # Étape finale
13
  FROM python:3.11
14
 
15
+ ENV CUDA_VISIBLE_DEVICES=""
16
+ ENV TF_FORCE_CPU_ALLOW_GROWTH=true
17
+ ENV PYTORCH_CUDA_ALLOC_CONF=max_split_size_mb:32
18
+
19
  WORKDIR /app
20
 
21
  # Créer le répertoire de cache et définir les permissions