Spaces:
Build error
Build error
dockerilfe
Browse files- Dockerfile +2 -1
Dockerfile
CHANGED
|
@@ -31,6 +31,7 @@ ENV PDM_CACHE_DIR $HOME/code/.pdm_cache/
|
|
| 31 |
ENV HF_HOME $HOME/code/.hf_home/
|
| 32 |
|
| 33 |
ENV DEVICE cuda:0
|
| 34 |
-
ENV ATTN_IMPLEMENTATION flash_attention_2
|
|
|
|
| 35 |
|
| 36 |
ENTRYPOINT [ "./entrypoint.sh" ]
|
|
|
|
| 31 |
ENV HF_HOME $HOME/code/.hf_home/
|
| 32 |
|
| 33 |
ENV DEVICE cuda:0
|
| 34 |
+
# ENV ATTN_IMPLEMENTATION flash_attention_2
|
| 35 |
+
ENV ATTN_IMPLEMENTATION flash_attention
|
| 36 |
|
| 37 |
ENTRYPOINT [ "./entrypoint.sh" ]
|