Spaces:
Runtime error
Runtime error
Update Dockerfile
Browse files- Dockerfile +1 -0
Dockerfile
CHANGED
|
@@ -6,6 +6,7 @@ COPY ./requirements.txt /code/requirements.txt
|
|
| 6 |
|
| 7 |
RUN pip install --upgrade -r /code/requirements.txt
|
| 8 |
|
|
|
|
| 9 |
RUN pip install flash-attn --no-build-isolation
|
| 10 |
|
| 11 |
RUN useradd -m -u 1000 user
|
|
|
|
| 6 |
|
| 7 |
RUN pip install --upgrade -r /code/requirements.txt
|
| 8 |
|
| 9 |
+
ENV FLASH_ATTENTION_SKIP_CUDA_BUILD=TRUE
|
| 10 |
RUN pip install flash-attn --no-build-isolation
|
| 11 |
|
| 12 |
RUN useradd -m -u 1000 user
|