roll-ai commited on
Commit
d970ea9
·
verified ·
1 Parent(s): fcb37d8

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -1
Dockerfile CHANGED
@@ -55,7 +55,7 @@ RUN pip install --no-cache-dir -r requirements.txt
55
  RUN pip uninstall -y flash-attn || true
56
 
57
  # Build from source so it links correctly
58
- RUN pip install flash-attn --no-build-isolation --no-cache-dir
59
 
60
  # Copy project code
61
  COPY . .
 
55
  RUN pip uninstall -y flash-attn || true
56
 
57
  # Build from source so it links correctly
58
+ RUN pip install flash-attn<2.8 --no-build-isolation --no-cache-dir
59
 
60
  # Copy project code
61
  COPY . .