Update entrypoint.sh

entrypoint.sh

@@ -39,24 +39,10 @@ echo ""
 WHEEL_DIR="/data/wheel"
 mkdir -p "$WHEEL_DIR"
 
-FLASH_ATTN_WHL=$(find "$WHEEL_DIR" -name "flash_attn-*.whl" | head -n 1)
+chmod +x /app/compile_flash.sh
+# Compila ou carrega FlashAttention do cache persistente
+bash /app/compile_flash.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..."