Spaces:
Paused
Paused
Update Dockerfile
Browse files- Dockerfile +5 -5
Dockerfile
CHANGED
|
@@ -67,7 +67,7 @@ RUN mkdir -p /home/user/.cache/models && ln -sf /home/user/.cache/models /app/mo
|
|
| 67 |
|
| 68 |
# ---------------- Sistema & Python ----------------
|
| 69 |
RUN apt-get update && apt-get install -y --no-install-recommends \
|
| 70 |
-
build-essential cmake git git-lfs curl wget ffmpeg ninja-build \
|
| 71 |
python3.10 python3.10-dev python3.10-distutils python3-pip \
|
| 72 |
&& apt-get clean && rm -rf /var/lib/apt/lists/*
|
| 73 |
|
|
@@ -75,7 +75,10 @@ RUN ln -sf /usr/bin/python3.10 /usr/bin/python3 && \
|
|
| 75 |
ln -sf /usr/bin/python3.10 /usr/bin/python && \
|
| 76 |
python3 -m pip install --upgrade pip
|
| 77 |
|
| 78 |
-
|
|
|
|
|
|
|
|
|
|
| 79 |
|
| 80 |
# FlashAttention 2.8.x
|
| 81 |
RUN pip install flash-attn==2.8.3 --no-build-isolation || \
|
|
@@ -113,9 +116,6 @@ 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 apt-get update && apt-get install -y \
|
| 117 |
-
python3-pybind11 ninja-build cmake gcc g++ python3-dev \
|
| 118 |
-
&& pip install --upgrade pip setuptools wheel pybind11 ninja cmake
|
| 119 |
|
| 120 |
RUN pip uninstall -y triton || true && \
|
| 121 |
pip install -v --no-build-isolation triton==3.4.0
|
|
|
|
| 67 |
|
| 68 |
# ---------------- Sistema & Python ----------------
|
| 69 |
RUN apt-get update && apt-get install -y --no-install-recommends \
|
| 70 |
+
build-essential cmake git git-lfs curl wget ffmpeg packaging ninja-build \
|
| 71 |
python3.10 python3.10-dev python3.10-distutils python3-pip \
|
| 72 |
&& apt-get clean && rm -rf /var/lib/apt/lists/*
|
| 73 |
|
|
|
|
| 75 |
ln -sf /usr/bin/python3.10 /usr/bin/python && \
|
| 76 |
python3 -m pip install --upgrade pip
|
| 77 |
|
| 78 |
+
RUN apt-get update && apt-get install -y \
|
| 79 |
+
python3-pybind11 ninja-build cmake gcc g++ python3-dev \
|
| 80 |
+
&& pip install --upgrade pip setuptools wheel pybind11 ninja cmake
|
| 81 |
+
|
| 82 |
|
| 83 |
# FlashAttention 2.8.x
|
| 84 |
RUN pip install flash-attn==2.8.3 --no-build-isolation || \
|
|
|
|
| 116 |
torch==2.8.0+cu128 torchvision==0.23.0+cu128 torchaudio==2.8.0+cu128
|
| 117 |
|
| 118 |
|
|
|
|
|
|
|
|
|
|
| 119 |
|
| 120 |
RUN pip uninstall -y triton || true && \
|
| 121 |
pip install -v --no-build-isolation triton==3.4.0
|