Update start.sh

start.sh

@@ -36,56 +36,57 @@ export NCCL_TIMEOUT=600
 if [ -d /data ]; then
   export HF_HOME="/data/.cache/huggingface"
   export TORCH_HOME="/data/.cache/torch"
-  mkdir -p "$HF_HOME/hub" "$TORCH_HOME"
-  # Symlink para compatibilidade com /app
-  mkdir -p /app/.cache
-  ln -sf "$HF_HOME" /app/.cache/huggingface
 else
   export HF_HOME="/app/.cache/huggingface"
   export TORCH_HOME="/app/.cache/torch"
-  mkdir -p "$HF_HOME/hub" "$TORCH_HOME"
 fi
-
-unset TRANSFORMERS_CACHE  # evitar conflito com outros caches
 export HF_HUB_CACHE="${HF_HUB_CACHE:-$HF_HOME/hub}"
+mkdir -p "$HF_HUB_CACHE" "$TORCH_HOME"
+
+# Symlink para compatibilidade com /app
+mkdir -p /app/.cache
+ln -sf "$HF_HOME" /app/.cache/huggingface
+
+unset TRANSFORMERS_CACHE
 export HF_HUB_ENABLE_HF_TRANSFER=1
 export HF_HUB_DOWNLOAD_TIMEOUT=60
 
-# ---------------------- Pré-download do modelo ----------------------
+# ---------------------- Pré-download do modelo (cache-first) ----------------------
 MODEL_REPO="ByteDance-Seed/VINCIE-3B"
-echo "Baixando o modelo ${MODEL_REPO} para cache persistente em $HF_HUB_CACHE..."
+CACHE_MODEL_DIR="$HF_HUB_CACHE/models--ByteDance-Seed--VINCIE-3B"
 
-retry_count=0
-max_retries=3
-while [ $retry_count -lt $max_retries ]; do
-  echo "Tentativa $((retry_count+1)) de snapshot_download..."
-  python - <<'PY' || true
-import os, time
+if [ -d "$CACHE_MODEL_DIR" ] && [ "$(ls -A "$CACHE_MODEL_DIR" 2>/dev/null)" ]; then
+  echo "Cache do modelo já existe em $CACHE_MODEL_DIR. Não será feito novo download."
+else
+  echo "Baixando o modelo ${MODEL_REPO} para cache persistente em $HF_HUB_CACHE..."
+  retry_count=0
+  max_retries=3
+  while [ $retry_count -lt $max_retries ]; do
+    echo "Tentativa $((retry_count+1)) de snapshot_download..."
+    python - <<'PY' || true
+import os
 from huggingface_hub import snapshot_download
 hf_home = os.environ.get("HF_HOME","/app/.cache/huggingface")
 cache_dir = os.path.join(hf_home, "hub")
 os.makedirs(cache_dir, exist_ok=True)
 repo_id = "ByteDance-Seed/VINCIE-3B"
 token = os.getenv("HF_TOKEN")
-try:
-  snapshot_download(repo_id=repo_id, cache_dir=cache_dir, resume_download=True, token=token)
-  print("Download concluído.")
-except Exception as e:
-  print(f"Erro no download: {e}")
-  raise
+snapshot_download(repo_id=repo_id, cache_dir=cache_dir, resume_download=True, token=token)
+print("Download concluído.")
 PY
-  status=$?
-  if [ $status -eq 0 ]; then
-    break
-  fi
-  retry_count=$((retry_count+1))
-  echo "Falha na tentativa $retry_count. Tentando novamente em 10s..."
-  sleep 10
-  if [ $retry_count -eq $max_retries ]; then
-    echo "Erro: Falha ao baixar o modelo após $max_retries tentativas."
-    exit 1
-  fi
-done
+    status=$?
+    if [ $status -eq 0 ]; then
+      break
+    fi
+    retry_count=$((retry_count+1))
+    echo "Falha na tentativa $retry_count. Tentando novamente em 10s..."
+    sleep 10
+    if [ $retry_count -eq $max_retries ]; then
+      echo "Erro: Falha ao baixar o modelo após $max_retries tentativas."
+      exit 1
+    fi
+  done
+fi
 
 ls -la "$HF_HUB_CACHE" || echo "Cache não encontrado após download."
 
@@ -93,7 +94,6 @@ ls -la "$HF_HUB_CACHE" || echo "Cache não encontrado após download."
 if nvidia-smi >/dev/null 2>&1; then
   if [ "${DISABLE_BUILDER:-0}" -eq 0 ]; then
     echo "Executando builder Apex/Q8..."
-    # Use o builder revisado (salve como /app/builder.sh)
     if [ -f /app/builder.sh ]; then
       chmod +x /app/builder.sh
       timeout "${BUILDER_TIMEOUT_SEC:-7200}" bash -lc /app/builder.sh || echo "Builder excedeu tempo/erro, prosseguindo."
@@ -116,7 +116,5 @@ echo "Estrutura em $HF_HOME:"
 ls -la "$HF_HOME" | sed -n '1,200p'
 
 # ---------------------- Subindo serviço ----------------------
-echo "🚀 Subindo serviços..."
-
-python app_vince.py
+python /app/app_vince.py