carlex3321 commited on
Commit
564ca3b
·
verified ·
1 Parent(s): 13ea254

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +2 -1
Dockerfile CHANGED
@@ -114,7 +114,8 @@ RUN pip install --index-url https://download.pytorch.org/whl/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
 
 
114
 
115
 
116
  RUN pip uninstall -y triton || true && \
117
+ pip install --pre --upgrade --extra-index-url https://download.pytorch.org/whl/nightly/cpu triton-nightly
118
+
119
 
120
  RUN pip install --upgrade bitsandbytes
121