Update Dockerfile

Dockerfile

@@ -88,10 +88,10 @@ RUN pip uninstall -y triton || true && \
     pip install -v --no-build-isolation triton==3.4.0
 
 # 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
@@ -100,12 +100,12 @@ RUN pip install --no-cache-dir diffusers==0.31.0 transformers==4.44.2 accelerate
 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  && \
-#   cp -r VINCIE/configs/. /app/configs/
+RUN git clone https://github.com/Dao-AILab/flash-attention  && \
+    cd flash-attention/csrc/layer_norm && pip install .
 
 
 RUN git clone https://github.com/bytedance-seed/VINCIE.git  &&  \
-    cd VINCIE && pip install -r requirements.txt && cd .. && \
+    #cd VINCIE && pip install -r requirements.txt && cd .. && \
     #pip install flash_attn==2.6.3 --no-build-isolation && \
     cp -r VINCIE/configs/. /app/configs/