Finish-him commited on
Commit
ef8f674
verified
1 Parent(s): 0c65912

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +4 -4
Dockerfile CHANGED
@@ -15,11 +15,11 @@ RUN pip install --no-cache-dir --upgrade -r requirements.txt
15
  # Define o caminho da cache
16
  ENV HF_HOME=/app/cache/huggingface
17
  ENV SENTENCE_TRANSFORMERS_HOME=/app/cache/torch
18
- RUN mkdir -p $HF_HOME && mkdir -p $SENTENCE_TRANSFORMERS_HOME
19
 
20
- # --- CORRE脟脙O PRINCIPAL: PR脡-AQUECIMENTO COM O MODELO e5-large ---
21
- # Este modelo muito mais leve e n茫o vai estourar a mem贸ria RAM.
22
- RUN python -c "from sentence_transformers import SentenceTransformer; SentenceTransformer('intfloat/multilingual-e5-large', cache_folder=os.environ.get('SENTENCE_TRANSFORMERS_HOME', '/app/cache/torch'))"
23
 
24
  # --- ETAPA 3: EXECUTAR O SCRIPT ---
25
  # Copia o script de treinamento
 
15
  # Define o caminho da cache
16
  ENV HF_HOME=/app/cache/huggingface
17
  ENV SENTENCE_TRANSFORMERS_HOME=/app/cache/torch
18
+ RUN mkdir -p $HF_HOME && mkdir -p $SENTENCE_TRANSFORMERS_HOME && chown -R 1000:1000 /app/cache
19
 
20
+ # --- CORRE脟脙O PRINCIPAL: PR脡-AQUECIMENTO DO CACHE ---
21
+ # Adiciona "import os;" ao in铆cio do comando python
22
+ RUN python -c "import os; from sentence_transformers import SentenceTransformer; SentenceTransformer('intfloat/multilingual-e5-large', cache_folder=os.environ.get('SENTENCE_TRANSFORMERS_HOME', '/app/cache/torch'))"
23
 
24
  # --- ETAPA 3: EXECUTAR O SCRIPT ---
25
  # Copia o script de treinamento