Update Dockerfile

Dockerfile

@@ -77,8 +77,6 @@ RUN pip install packaging ninja cmake pybind11 scikit-build cython hf_transfer "
 # --- Triton 3.x ---
 RUN pip uninstall -y triton || true && \
     pip install -v --no-build-isolation triton==3.4.0
-
-# --- FlashAttention 2.8.x ---
   
 # =============================================================================
 # 4. Instalação das Dependências da Aplicação
@@ -98,6 +96,7 @@ RUN pip install --no-cache-dir \
 
 RUN echo "Installing custom wheels..." && \
     pip install --no-cache-dir \
+      "https://huggingface.co/caarleexx/Flash/resolve/main/flash_attn-2.8.0.post2-cp310-cp310-linux_x86_64.whl" \
       "https://huggingface.co/euIaxs22/Aduc-sdr/resolve/main/apex-0.1-cp310-cp310-linux_x86_64.whl" \
       "https://huggingface.co/euIaxs22/Aduc-sdr/resolve/main/dropout_layer_norm-0.1-cp310-cp310-linux_x86_64.whl"