Spaces:
Paused
Paused
Update Dockerfile
Browse files- Dockerfile +2 -2
Dockerfile
CHANGED
|
@@ -57,7 +57,7 @@ RUN pip install --index-url https://download.pytorch.org/whl/cu128 \
|
|
| 57 |
torch==2.8.0+cu128 torchvision==0.23.0+cu128 torchaudio==2.8.0+cu128
|
| 58 |
|
| 59 |
# ---------------- Toolchain, Triton, FA2 (sem bnb) ----------------
|
| 60 |
-
RUN pip install packaging ninja cmake pybind11 scikit-build cython hf_transfer numpy
|
| 61 |
|
| 62 |
# Triton 3.x (sem triton.ops)
|
| 63 |
RUN pip uninstall -y triton || true && \
|
|
@@ -70,7 +70,7 @@ RUN pip install flash-attn==2.8.3 --no-build-isolation || \
|
|
| 70 |
pip install flash-attn==2.8.0.post2 --no-build-isolation
|
| 71 |
|
| 72 |
# Diffusers/Transformers estáveis (sem dev)
|
| 73 |
-
RUN pip install --no-cache-dir diffusers
|
| 74 |
|
| 75 |
# Opcional: seu fork de otimizações
|
| 76 |
RUN pip install -U git+https://github.com/carlex22/diffusers-aduc-sdr
|
|
|
|
| 57 |
torch==2.8.0+cu128 torchvision==0.23.0+cu128 torchaudio==2.8.0+cu128
|
| 58 |
|
| 59 |
# ---------------- Toolchain, Triton, FA2 (sem bnb) ----------------
|
| 60 |
+
RUN pip install packaging ninja cmake pybind11 scikit-build cython hf_transfer numpy>=1.24.4
|
| 61 |
|
| 62 |
# Triton 3.x (sem triton.ops)
|
| 63 |
RUN pip uninstall -y triton || true && \
|
|
|
|
| 70 |
pip install flash-attn==2.8.0.post2 --no-build-isolation
|
| 71 |
|
| 72 |
# Diffusers/Transformers estáveis (sem dev)
|
| 73 |
+
RUN pip install --no-cache-dir diffusers>=0.31.0 transformers>=4.44.2 accelerate>=0.34.2 omegaconf>=2.3.0
|
| 74 |
|
| 75 |
# Opcional: seu fork de otimizações
|
| 76 |
RUN pip install -U git+https://github.com/carlex22/diffusers-aduc-sdr
|