Files
modelscope/docker/scripts/install_flash_attension.sh
2023-09-01 18:08:35 +08:00

7 lines
231 B
Bash

git clone -b v1.0.8 https://github.com/Dao-AILab/flash-attention && \
cd flash-attention && pip install . && \
pip install csrc/layer_norm && \
pip install csrc/rotary && \
cd .. && \
rm -rf flash-attention