Spaces:
Paused
Paused
Update Dockerfile
Browse files- Dockerfile +2 -1
Dockerfile
CHANGED
|
@@ -19,7 +19,8 @@ RUN pip install --upgrade pip && \
|
|
| 19 |
pip install --no-cache-dir -r requirements.txt
|
| 20 |
|
| 21 |
# Optional: flash‑attn (Ampere 86 wird erkannt, Wheel vorhanden)
|
| 22 |
-
|
|
|
|
| 23 |
|
| 24 |
# Mount das geheime HF‑Token beim Build:
|
| 25 |
# Settings → Secrets → Name: HF_TOKEN (scope: "read")
|
|
|
|
| 19 |
pip install --no-cache-dir -r requirements.txt
|
| 20 |
|
| 21 |
# Optional: flash‑attn (Ampere 86 wird erkannt, Wheel vorhanden)
|
| 22 |
+
# fertiges Wheel laden (kein nvcc)
|
| 23 |
+
RUN pip install --no-cache-dir "flash-attn>=2.6.3,<=2.7" --extra-index-url https://download.pytorch.org/whl/cu118
|
| 24 |
|
| 25 |
# Mount das geheime HF‑Token beim Build:
|
| 26 |
# Settings → Secrets → Name: HF_TOKEN (scope: "read")
|