Update README.md

This commit is contained in:
Junyang Lin
2023-10-08 15:35:07 +08:00
committed by GitHub
parent 98289a03b0
commit c7cf15dbdc

View File

@@ -102,7 +102,7 @@ pip install -r requirements.txt
If your device supports fp16 or bf16, we recommend installing [flash-attention](https://github.com/Dao-AILab/flash-attention) (**we support flash attention 2 now.**) for higher efficiency and lower memory usage. (**flash-attention is optional and the project can run normally without installing it**) If your device supports fp16 or bf16, we recommend installing [flash-attention](https://github.com/Dao-AILab/flash-attention) (**we support flash attention 2 now.**) for higher efficiency and lower memory usage. (**flash-attention is optional and the project can run normally without installing it**)
```bash ```bash
# Previous installation commands. Now Flash Attention 2 is supported. # Previous installation commands. Now flash attention 2 is supported.
# git clone -b v1.0.8 https://github.com/Dao-AILab/flash-attention # git clone -b v1.0.8 https://github.com/Dao-AILab/flash-attention
# cd flash-attention && pip install . # cd flash-attention && pip install .
pip install flash-attn --no-build-isolation pip install flash-attn --no-build-isolation