Update start.sh

start.sh

@@ -5,7 +5,6 @@ echo "======================================================="
 echo " ADUC-SDR — Start (VINCIE/SeedVR, 8× L40S)"
 echo "======================================================="
 
-
 # ---------------------- Env base ----------------------
 export CUDA_VISIBLE_DEVICES="${CUDA_VISIBLE_DEVICES:-0,1,2,3,4,5,6,7}"
 export TORCH_DTYPE="${TORCH_DTYPE:-bfloat16}"
@@ -37,31 +36,32 @@ export NCCL_TIMEOUT=600
 export HF_HOME=/app/.cache/huggingface
 unset TRANSFORMERS_CACHE
 
-
-df -h  
-# Mostra uso de disco por volume (procure /app ou o filesystem raiz).
-du -sh /app/*  
-# Mostra quanto cada subdiretório em /app está usando.
-
-
-
 # ---------------------- Baixar o modelo antes de iniciar ----------------------
-echo "Baixando o modelo ByteDance-Seed/VINCIE-3B para cache persistente..."
-
-# Use Python para baixar o modelo via huggingface_hub
-python -c "
-import os
-from huggingface_hub import snapshot_download
-
-os.environ['HF_HOME'] = '/app/.cache/huggingface'
-repo_id = 'ByteDance-Seed/VINCIE-3B'
-snapshot_download(repo_id=repo_id, cache_dir=os.path.join(os.environ['HF_HOME'], 'hub'))
-" || { echo "Erro ao baixar o modelo. Continuando..."; }
-
-# Verificar se o cache foi criado
-ls -la $HF_HOME/hub || echo "Cache não encontrado após download."
-
+echo "Criando diretório de cache se não existir..."
+mkdir -p /app/.cache/huggingface/hub
+
+# Ativando hf_transfer para downloads mais rápidos
+export HF_HUB_ENABLE_HF_TRANSFER=1
+export HF_HUB_DOWNLOAD_TIMEOUT=60
+
+MODEL_REPO="ByteDance-Seed/VINCIE-3B"
+
+retry_count=0
+max_retries=3
+
+while [ $retry_count -lt $max_retries ]; do
+  echo "Tentativa $((retry_count+1)) de baixar o modelo $MODEL_REPO..."
+  python -c "import os; from huggingface_hub import snapshot_download; os.environ['HF_HOME']='/app/.cache/huggingface'; from os import getenv; snapshot_download(repo_id='$MODEL_REPO', cache_dir=os.path.join(os.environ['HF_HOME'], 'hub'), resume_download=True, token=getenv('HF_TOKEN'))" && break
+  retry_count=$((retry_count+1))
+  echo "Falha na tentativa $retry_count. Tentando novamente em 10 segundos..."
+  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
 
+ls -la /app/.cache/huggingface/hub || echo "Cache não encontrado após download."
 
 # ---------------------- Builder Apex/Q8 ----------------------
 if nvidia-smi >/dev/null 2>&1; then
@@ -81,10 +81,16 @@ else
   echo "GPU não visível, pulando builder Apex/Q8."
 fi
 
-# Configuráveis
+# ---------------------- Banner ----------------------
 ./info.sh || true
 
-echo "Subindo serviços..."
-# ---------------------- Banner ----------------------
-python /app/app_vince.py
+
+ls -la /app && ls -R /app | head -n 2000
+
+echo "🚀 Subindo serviços..."
+
+# Exemplo: subir UI mínima SD Img2Img (ajuste conforme seu app)
+# python app_animatediff_min.py
+
 # Ou subir VINCIE UI se for o caso
+python app_vince.py