Spaces:
Paused
Paused
chore(docker): try pip install flash-attn fallback when wheel tag mismatches
Browse files- Dockerfile +2 -1
Dockerfile
CHANGED
|
@@ -73,7 +73,8 @@ RUN if [ "$INSTALL_FLASH_ATTN" = "true" ]; then \
|
|
| 73 |
echo "Installing flash-attn from $FLASH_ATTN_WHEEL_URL" && \
|
| 74 |
pip install --no-cache-dir "$FLASH_ATTN_WHEEL_URL"; \
|
| 75 |
else \
|
| 76 |
-
echo "flash-attn wheel tag mismatch for $PYTAG.
|
|
|
|
| 77 |
fi; \
|
| 78 |
else \
|
| 79 |
echo "Skipping flash-attn installation"; \
|
|
|
|
| 73 |
echo "Installing flash-attn from $FLASH_ATTN_WHEEL_URL" && \
|
| 74 |
pip install --no-cache-dir "$FLASH_ATTN_WHEEL_URL"; \
|
| 75 |
else \
|
| 76 |
+
echo "flash-attn wheel tag mismatch for $PYTAG. Trying pip install flash-attn (no-build-isolation)." && \
|
| 77 |
+
(pip install --no-cache-dir --no-build-isolation flash-attn || echo "flash-attn pip install failed; continuing without flash-attn"); \
|
| 78 |
fi; \
|
| 79 |
else \
|
| 80 |
echo "Skipping flash-attn installation"; \
|