Tomtom84 commited on
Commit
efa5012
·
verified ·
1 Parent(s): 9ecbb7d

Update Dockerfile

Browse files
Files changed (1) hide show
  1. 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
- RUN pip install --no-cache-dir flash-attn==2.5.2 --no-build-isolation
 
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")