Spaces:
Paused
Paused
Update Dockerfile
Browse files- 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.
|
| 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 |
|