Update Dockerfile

Dockerfile

@@ -90,6 +90,15 @@ RUN pip install --no-cache-dir -r requirements.txt
 # --- Instalação de bitsandbytes e Wheels Customizados (Mantido 100% Original) ---
 RUN pip install --upgrade bitsandbytes
 
+
+# --- FlashAttention 2.8.x ---
+RUN pip install flash-attn==2.8.3 --no-build-isolation || \
+    pip install flash-attn==2.8.2 --no-build-isolation || \
+    pip install flash-attn==2.8.1 --no-build-isolation || \
+    pip install flash-attn==2.8.0.post2 --no-build-isolation
+
+
+
 # Instala wheels customizados (Apex, etc.)
 # Instala q8_kernels
 RUN pip install --no-cache-dir \
@@ -97,9 +106,11 @@ RUN pip install --no-cache-dir \
 
 RUN echo "Installing custom wheels..." && \
     pip install --no-cache-dir \
-      "https://huggingface.co/carlex3321/aduc-sdr3/resolve/main/flash_attn-2.6.3-cp310-cp310-linux_x86_64.whl" \
+      #"https://huggingface.co/carlex3321/aduc-sdr3/resolve/main/flash_attn-2.6.3-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" 
+
+
       
 # =============================================================================
 # 5. Cópia do Código-Fonte e Configuração Final