Update Dockerfile

Dockerfile

@@ -67,7 +67,7 @@ RUN mkdir -p /home/user/.cache/models && ln -sf /home/user/.cache/models /app/mo
 
 # ---------------- Sistema & Python ----------------
 RUN apt-get update && apt-get install -y --no-install-recommends \
-    build-essential cmake git git-lfs curl wget ffmpeg packaging ninja-build \
+    build-essential cmake git git-lfs curl wget ffmpeg ninja-build \
     python3.10 python3.10-dev python3.10-distutils python3-pip \
     && apt-get clean && rm -rf /var/lib/apt/lists/*
 
@@ -75,10 +75,17 @@ RUN ln -sf /usr/bin/python3.10 /usr/bin/python3 && \
     ln -sf /usr/bin/python3.10 /usr/bin/python && \
     python3 -m pip install --upgrade pip
 
-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
-    
+
+# ---------------- PyTorch cu128 (pinado) ----------------
+RUN pip install --index-url https://download.pytorch.org/whl/cu128 \
+    torch==2.8.0+cu128 torchvision==0.23.0+cu128 torchaudio==2.8.0+cu128
+
+# ---------------- Toolchain, Triton, FA2 (sem bnb) ----------------
+RUN pip install packaging ninja cmake pybind11 scikit-build cython hf_transfer numpy==1.24.4
+
+# Triton 3.x (sem triton.ops)
+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 || \
@@ -90,7 +97,7 @@ RUN pip install flash-attn==2.8.3 --no-build-isolation || \
 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  && \
@@ -111,19 +118,8 @@ COPY requirements.txt ./requirements.txt
 RUN pip install --no-cache-dir -r requirements.txt
 
 
-# ---------------- PyTorch cu128 (pinado) ----------------
-RUN pip install --index-url https://download.pytorch.org/whl/cu128 \
-    torch==2.8.0+cu128 torchvision==0.23.0+cu128 torchaudio==2.8.0+cu128
-
-
-
-RUN pip uninstall -y triton || true && \
-    pip install -v --no-build-isolation triton==3.4.0
-
-
 RUN pip install --upgrade bitsandbytes
 
-
 # Scripts e configs
 COPY info.sh ./info.sh
 COPY start.sh ./start.sh
@@ -146,10 +142,9 @@ ENV TORCH_HOME=/data/.cache/torch
 ENV HF_DATASETS_CACHE=/data/.cache/datasets
 ENV TRANSFORMERS_CACHE=/data/.cache/transformers
 ENV DIFFUSERS_CACHE=/data/.cache/diffusers
-ENV FLASH_ATTENTION_TRITON_AMD_ENABLE="TRUE"
 
 
 
 # ---------------- Entry ----------------
 ENTRYPOINT ["./start.sh"]
-CMD ["gradio"]
+CMD ["gradio"]