Update entrypoint.sh

entrypoint.sh

@@ -3,10 +3,6 @@ set -e
 
 echo "🚀 ADUC-SDR Entrypoint: Configurando o ambiente de execução..."
 
-
-
-
-
 # --- Configuração de Performance (CPU & GPU) ---
 NUM_VCPUS=$(nproc)
 NUM_GPUS=$(nvidia-smi --query-gpu=count --format=csv,noheader | head -n 1 || echo 0)
@@ -37,11 +33,30 @@ echo "   > Depuração: ADUC_LOG_LEVEL=${ADUC_LOG_LEVEL}, CUDA_LAUNCH_BLOCKING=$
 echo ""
 echo ""
 
+# =============================================================================
+# 🔧 Compilação e Cache de Wheels (FlashAttention e outros)
+# =============================================================================
+WHEEL_DIR="/data/wheel"
+mkdir -p "$WHEEL_DIR"
 
-pip install flash-attn==2.8.3 --no-build-isolation
+FLASH_ATTN_WHL=$(find "$WHEEL_DIR" -name "flash_attn-*.whl" | head -n 1)
 
-
-#/bin/bash /app/info.sh
+if [[ -f "$FLASH_ATTN_WHL" ]]; then
+    echo "📦 Encontrado wheel compilado: $FLASH_ATTN_WHL"
+    echo "   > Instalando diretamente sem recompilar..."
+    pip install --no-cache-dir "$FLASH_ATTN_WHL"
+else
+    echo "⚙️ Nenhum wheel encontrado. Compilando flash-attn..."
+    pip wheel flash-attn==2.8.3 --no-build-isolation -w "$WHEEL_DIR"
+    FLASH_ATTN_WHL=$(find "$WHEEL_DIR" -name "flash_attn-*.whl" | head -n 1)
+    if [[ -f "$FLASH_ATTN_WHL" ]]; then
+        echo "✅ Compilação concluída. Instalando a partir do cache local."
+        pip install --no-cache-dir "$FLASH_ATTN_WHL"
+    else
+        echo "❌ Falha ao compilar flash-attn."
+        exit 1
+    fi
+fi
 
 # --- Setup de Dependências ---
 echo "   > Verificando dependências com setup.py..."
@@ -49,4 +64,4 @@ python3 /app/setup.py
 
 echo "---------------------------------------------------------"
 echo "🔥 Ambiente configurado. Iniciando o comando principal: $@"
-exec "$@"
+exec "$@"