carlex3321 commited on
Commit
8e139e3
·
verified ·
1 Parent(s): 687a9ef

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -4
Dockerfile CHANGED
@@ -112,12 +112,9 @@ RUN pip install --no-cache-dir -r requirements.txt
112
  RUN pip install --index-url https://download.pytorch.org/whl/cu128 \
113
  torch==2.8.0+cu128 torchvision==0.23.0+cu128 torchaudio==2.8.0+cu128
114
 
115
- # ---------------- Toolchain, Triton, FA2 (sem bnb) ----------------
116
- RUN pip install packaging ninja cmake pybind11 scikit-build cython hf_transfer numpy==1.24.4
117
 
118
- # Triton 3.x (sem triton.ops)
119
  RUN pip uninstall -y triton || true && \
120
- pip install -v --no-build-isolation triton==3.1.0
121
 
122
  RUN pip install --upgrade bitsandbytes
123
 
 
112
  RUN pip install --index-url https://download.pytorch.org/whl/cu128 \
113
  torch==2.8.0+cu128 torchvision==0.23.0+cu128 torchaudio==2.8.0+cu128
114
 
 
 
115
 
 
116
  RUN pip uninstall -y triton || true && \
117
+ pip install -v --no-build-isolation triton==3.4.0
118
 
119
  RUN pip install --upgrade bitsandbytes
120