carlex3321 commited on
Commit
31010a8
·
verified ·
1 Parent(s): bd40b35

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +2 -2
Dockerfile CHANGED
@@ -117,9 +117,9 @@ RUN pip install packaging ninja cmake pybind11 scikit-build cython hf_transfer n
117
 
118
  # Triton 3.x (sem triton.ops)
119
  RUN pip uninstall -y triton || true && \
120
- pip install -v --no-build-isolation triton==3.2.0
121
-
122
 
 
123
 
124
 
125
  # Scripts e configs
 
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
 
124
 
125
  # Scripts e configs