Update compile_flash.sh

compile_flash.sh

@@ -30,13 +30,13 @@ echo "   > TMPDIR=$TMPDIR, WHEEL_DIR=$WHEEL_DIR"
 
 # Tenta build com CUDA
 set -o pipefail
-if pip wheel flash-attn==2.8.0 --no-build-isolation -w "$WHEEL_DIR" 2>&1 | tee /tmp/flash_build.log; then
+if pip wheel flash-attn==2.8.0.post2 --no-build-isolation -w "$WHEEL_DIR" 2>&1 | tee /tmp/flash_build.log; then
   echo "✅ Build com CUDA concluído."
 else
   echo "⚠️ Build CUDA falhou; tentando fallback CPU-only (mais lento). Verifique /tmp/flash_build.log"
   export USE_FLASH_ATTENTION_WITH_CUDA=0
   export FORCE_CUDA=0
-  pip wheel flash-attn==2.8.0 --no-build-isolation -w "$WHEEL_DIR"
+  pip wheel flash-attn==2.8.0.post2 --no-build-isolation -w "$WHEEL_DIR"
 fi
 
 FLASH_WHL=$(find "$WHEEL_DIR" -name "flash_attn-*.whl" | head -n 1)