Update Dockerfile
Browse files- Dockerfile +4 -4
Dockerfile
CHANGED
|
@@ -79,10 +79,10 @@ RUN pip uninstall -y triton || true && \
|
|
| 79 |
pip install -v --no-build-isolation triton==3.4.0
|
| 80 |
|
| 81 |
# --- FlashAttention 2.8.x ---
|
| 82 |
-
RUN pip install flash-attn==2.8.3 --no-build-isolation || \
|
| 83 |
-
pip install flash-attn==2.8.2 --no-build-isolation || \
|
| 84 |
-
pip install flash-attn==2.8.1 --no-build-isolation || \
|
| 85 |
-
pip install flash-attn==2.8.0.post2 --no-build-isolation
|
| 86 |
|
| 87 |
# =============================================================================
|
| 88 |
# 4. Instalação das Dependências da Aplicação
|
|
|
|
| 79 |
pip install -v --no-build-isolation triton==3.4.0
|
| 80 |
|
| 81 |
# --- FlashAttention 2.8.x ---
|
| 82 |
+
#RUN pip install flash-attn==2.8.3 --no-build-isolation || \
|
| 83 |
+
# pip install flash-attn==2.8.2 --no-build-isolation || \
|
| 84 |
+
# pip install flash-attn==2.8.1 --no-build-isolation || \
|
| 85 |
+
# pip install flash-attn==2.8.0.post2 --no-build-isolation
|
| 86 |
|
| 87 |
# =============================================================================
|
| 88 |
# 4. Instalação das Dependências da Aplicação
|