Finish-him commited on
Commit
d8166a5
·
verified ·
1 Parent(s): 9edd46d

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +2 -2
Dockerfile CHANGED
@@ -18,8 +18,8 @@ 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
- # Força o download do modelo durante a fase de build, que tem um timeout maior.
22
- RUN python -c "from sentence_transformers import SentenceTransformer; SentenceTransformer('intfloat/e5-mistral-7b-instruct', cache_folder=os.environ.get('SENTENCE_TRANSFORMERS_HOME', '/app/cache/torch'), trust_remote_code=True)"
23
 
24
  # --- ETAPA 3: EXECUTAR O SCRIPT ---
25
  # Copia o script de treinamento
 
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/e5-mistral-7b-instruct', cache_folder=os.environ.get('SENTENCE_TRANSFORMERS_HOME', '/app/cache/torch'), trust_remote_code=True)"
23
 
24
  # --- ETAPA 3: EXECUTAR O SCRIPT ---
25
  # Copia o script de treinamento