Update start.sh

start.sh

@@ -51,28 +51,54 @@ export HF_HUB_ENABLE_HF_TRANSFER=1
 export HF_HUB_DOWNLOAD_TIMEOUT=60
 
 MODEL_REPO="ByteDance-Seed/VINCIE-3B"
-CACHE_MODEL_DIR="$HF_HUB_CACHE/models--ByteDance-Seed--VINCIE-3B"
 CKPT_DIR="/app/ckpt/VINCIE-3B"
 mkdir -p "$CKPT_DIR"
 
-# ---------------------- Download: cache estruturado ou direto ----------------------
+# ---------------------- Cache Estruturado HF (persistente) ----------------------
 
+# Define cache no volume persistente /data (1TB)
+if [ -d /data ]; then
+  export HF_HOME="${HF_HOME:-/data/.cache/huggingface}"
+else
+  export HF_HOME="${HF_HOME:-/app/.cache/huggingface}"
+fi
+
+export HF_HUB_CACHE="${HF_HUB_CACHE:-$HF_HOME/hub}"
+mkdir -p "$HF_HUB_CACHE"
+
+echo "📦 Cache HF: $HF_HUB_CACHE"
+
+# Download usando cache estruturado (não duplica arquivos)
 python3 - <<'PY'
 from huggingface_hub import snapshot_download
 import os
-save_dir = '/app/ckpt/VINCIE-3B'
-os.makedirs(save_dir, exist_ok=True)
+
+cache_dir = os.environ.get('HF_HUB_CACHE')
+print(f'📥 Baixando VINCIE-3B para cache: {cache_dir}')
+
 try:
-    print('📥 Baixando VINCIE-3B...')
-    snapshot_download(
+    model_path = snapshot_download(
         repo_id='ByteDance-Seed/VINCIE-3B',
-        local_dir=save_dir,
-        resume_download=True,
-        local_dir_use_symlinks=False
+        cache_dir=cache_dir,           # Usa cache estruturado
+        resume_download=True,          # Retoma downloads interrompidos
+        max_workers=8,                 # Acelera com paralelismo
+        # Não usa local_dir - mantém tudo no cache HF
     )
-    print('✅ Modelo ok')
+    print(f'✅ Modelo em cache: {model_path}')
+    
+    # Cria symlink para compatibilidade com código legacy
+    ckpt_link = '/app/ckpt/VINCIE-3B'
+    os.makedirs('/app/ckpt', exist_ok=True)
+    if os.path.islink(ckpt_link):
+        os.unlink(ckpt_link)
+    if not os.path.exists(ckpt_link):
+        os.symlink(model_path, ckpt_link)
+        print(f'🔗 Symlink: {ckpt_link} -> {model_path}')
+        
 except Exception as e:
     print(f'⚠️ Download falhou: {e}')
+    import traceback
+    traceback.print_exc()
 PY
 
 
@@ -91,10 +117,6 @@ else
 fi
 
 
-# Triton 3.x (sem triton.ops)
-pip uninstall -y triton || true && \
-    pip install -v --no-build-isolation triton==3.2.0
-
 
 # ---------------------- Diagnóstico ----------------------
 /app/info.sh || true