euIaxs22 commited on
Commit
3561d5e
·
verified ·
1 Parent(s): c07a9c9

Update Dockerfile

Browse files
Files changed (1) hide show
  1. 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==1.24.4
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==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
 
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