Update Dockerfile

Dockerfile

@@ -78,16 +78,16 @@ RUN ln -sf /usr/bin/python3.10 /usr/bin/python3 && \
 
 
 # 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
+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
 
 # Diffusers/Transformers estáveis (sem dev)
 RUN pip install --no-cache-dir diffusers==0.31.0 transformers==4.44.2 accelerate==0.34.2 omegaconf==2.3.0
 
 # Opcional: seu fork de otimizações
-# RUN pip install -U git+https://github.com/carlex22/diffusers-aduc-sdr
+RUN pip install -U git+https://github.com/carlex22/diffusers-aduc-sdr
 
 # ---------------- Repositórios auxiliares ----------------
 RUN git clone https://github.com/bytedance-seed/VINCIE.git  && \
@@ -117,7 +117,8 @@ RUN apt-get update && apt-get install -y \
     python3-pybind11 ninja-build cmake gcc g++ python3-dev \
     && pip install --upgrade pip setuptools wheel pybind11 ninja cmake
 
-
+RUN pip uninstall -y triton || true && \
+    pip install -v --no-build-isolation triton==3.4.0
 
 
 RUN pip install --upgrade bitsandbytes