Julian Bilcke
commited on
Commit
·
f08804f
1
Parent(s):
6ea6b16
update Dockerfile
Browse files- Dockerfile +6 -0
Dockerfile
CHANGED
|
@@ -36,6 +36,12 @@ ENV PYTHONPATH=$HOME/app \
|
|
| 36 |
PYTHONUNBUFFERED=1 \
|
| 37 |
DATA_ROOT=/tmp/data
|
| 38 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 39 |
RUN echo "Installing requirements.txt"
|
| 40 |
RUN pip3 install --no-cache-dir --upgrade -r /code/requirements.txt
|
| 41 |
|
|
|
|
| 36 |
PYTHONUNBUFFERED=1 \
|
| 37 |
DATA_ROOT=/tmp/data
|
| 38 |
|
| 39 |
+
RUN echo "Installing Ninja"
|
| 40 |
+
RUN pip3 install ninja
|
| 41 |
+
|
| 42 |
+
RUN echo "Building Flash Attention"
|
| 43 |
+
RUN pip3 install flash-attn --no-build-isolation
|
| 44 |
+
|
| 45 |
RUN echo "Installing requirements.txt"
|
| 46 |
RUN pip3 install --no-cache-dir --upgrade -r /code/requirements.txt
|
| 47 |
|