Update Dockerfile

Dockerfile

@@ -83,38 +83,18 @@ RUN pip install --index-url https://download.pytorch.org/whl/cu128 \
 # ---------------- Toolchain, Triton, FA2 (sem bnb) ----------------
 RUN pip install packaging ninja cmake pybind11 scikit-build cython hf_transfer numpy==1.24.4
 
-
-
-# ---------------- Toolchain base ----------------
-RUN pip install packaging ninja cmake pybind11 scikit-build cython hf_transfer numpy==1.24.4
-
-# ---------------- Instalação via wheels (HF Hub) ----------------
-# Triton, Apex e Q8 pré-compilados (Python 3.10, manylinux x86_64)
-# Use sempre --no-cache-dir e --no-build-isolation para evitar recompilar
+# Triton 3.x (sem triton.ops)
 RUN pip uninstall -y triton || true && \
-    pip install --no-cache-dir --no-build-isolation \
-      https://huggingface.co/carlex3321/aduc-sdr/resolve/main/triton-3.5.0%2Bgit51021fb2-cp310-cp310-linux_x86_64.whl && \
-    pip install --no-cache-dir --no-build-isolation \
-      https://huggingface.co/carlex3321/aduc-sdr/resolve/main/apex-0.1-cp310-cp310-linux_x86_64.whl
-    #pip install --no-cache-dir --no-build-isolation \
-    #  https://huggingface.co/carlex3321/aduc-sdr/resolve/main/q8_kernels-0.0.5-cp310-cp310-linux_x86_64.whl
-
-# ---------------- FlashAttention 2.8.x ----------------
-# Mantém fallback em cascata caso a versão mais nova falhe
-RUN pip install --no-cache-dir flash-attn==2.8.3 --no-build-isolation || \
-    pip install --no-cache-dir flash-attn==2.8.2 --no-build-isolation || \
-    pip install --no-cache-dir flash-attn==2.8.1 --no-build-isolation || \
-    pip install --no-cache-dir flash-attn==2.8.0.post2 --no-build-isolation
-    
+    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 transformers==4.44.2 accelerate==0.34.2 omegaconf==2.3.0
+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
@@ -138,12 +118,6 @@ COPY requirements.txt ./requirements.txt
 RUN pip install --no-cache-dir -r requirements.txt
 
 
-RUN apt-get update && apt-get install -y \
-    build-essential clang lld cmake ninja-build git python3-dev \
- && rm -rf /var/lib/apt/lists/*
-ENV CC=clang CXX=clang++ LD=lld
-
-
 RUN pip install --upgrade bitsandbytes
 
 # Scripts e configs
@@ -170,7 +144,6 @@ ENV TRANSFORMERS_CACHE=/data/.cache/transformers
 ENV DIFFUSERS_CACHE=/data/.cache/diffusers
 
 
-
 # ---------------- Entry ----------------
 ENTRYPOINT ["./start.sh"]
 CMD ["gradio"]