JoseMSL commited on
Commit
3401387
·
verified ·
1 Parent(s): 90fed95

Upload Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +30 -9
Dockerfile CHANGED
@@ -10,12 +10,33 @@ COPY . .
10
  ENV PORT=7860
11
  EXPOSE 7860
12
 
13
- # NOTA v8-intent (joint model): difiere del v6/v7-part6 en 2 flags
14
- # --workers 1 (vs 2): el joint model bloquea ~1 vCPU completo por inferencia (GIL).
15
- # CPU Basic free tiene 2 vCPU. Con 2 workers, ambos pelean por CPU+RAM en
16
- # paralelo y la primera inferencia post-fork excedía los 120s del timeout,
17
- # causando WORKER TIMEOUT → SIGKILL. Con 1 worker hay margen sobrado.
18
- # --timeout 300 (vs 120): margen para primera inferencia post-fork con
19
- # DistilBERT-multilingual + load state_dict + 2 heads + lazy kernel init.
20
- # Inferencias posteriores son rápidas (~600-800ms warm).
21
- CMD ["gunicorn", "--bind", "0.0.0.0:7860", "--workers", "1", "--preload", "--timeout", "300", "bert_api:app"]
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
10
  ENV PORT=7860
11
  EXPOSE 7860
12
 
13
+ # PYTHONUNBUFFERED=1 fuerza el flush inmediato de stdout/stderr en Python.
14
+ # Sin esto, los print() del bert_api.py (carga del modelo, warmup, etc.) quedan
15
+ # en buffer y NO aparecen en los Container logs de HF Spaces antes de SIGKILL.
16
+ # Documentado en doc 40 §11.10 y en HF forum (foros.huggingface.co).
17
+ ENV PYTHONUNBUFFERED=1
18
+
19
+ # NOTA v8-intent (joint model): difiere del v6/v7-part6 en 3 flags críticos.
20
+ # Causa raíz documentada en doc 40 §11.10 — bug conocido fork+OpenMP+PyTorch.
21
+ #
22
+ # ❌ NO usar --preload con modelos PyTorch custom (JointIntentSlotModel).
23
+ # Bug pytorch/pytorch#49555 + gunicorn#2478: PyTorch inicializa su thread
24
+ # pool intra-op (OpenMP/MKL) en master. Tras fork(), los workers heredan
25
+ # el estado del runtime OpenMP pero NO los threads → deadlock infinito en
26
+ # el primer model.forward(). Síntoma: WORKER TIMEOUT exactamente al
27
+ # --timeout configurado.
28
+ # Solución (Opción A doc 40 §11.10): cargar el modelo en cada worker
29
+ # post-fork. Sin master pre-fork, sin OpenMP heredado, sin deadlock.
30
+ #
31
+ # --workers 1: el joint model bloquea ~1 vCPU por GIL en cada inferencia.
32
+ # CPU Basic free tiene 2 vCPU. Con workers=1 hay margen y el modelo se
33
+ # carga UNA vez (sin --preload se cargaría por cada worker → +RAM si
34
+ # escaláramos workers).
35
+ #
36
+ # --timeout 120: vuelta al default del v6/v7-part6. Sin el deadlock OpenMP,
37
+ # las inferencias son ~600-800ms warm. No hace falta el --timeout 300
38
+ # excepcional.
39
+ #
40
+ # Trade-off conocido: arranque del worker tarda ~30-60s (carga el modelo en
41
+ # worker en lugar de heredarlo de master). Inferencias warm sin cambio.
42
+ CMD ["gunicorn", "--bind", "0.0.0.0:7860", "--workers", "1", "--timeout", "120", "bert_api:app"]