Skip to content

Commit

Permalink
Added pip install ninja to accelerate installation of flash-attn (#1461)
Browse files Browse the repository at this point in the history
* Added pip install ninja to accelerate installation of flash-attn

* doc: cleanup
  • Loading branch information
melvinebenezer authored Apr 2, 2024
1 parent 586bd8d commit cae608f
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions README.md
Original file line number Diff line number Diff line change
@@ -107,7 +107,7 @@ Get started with Axolotl in just a few steps! This quickstart guide will walk yo
git clone https://github.com/OpenAccess-AI-Collective/axolotl
cd axolotl

pip3 install packaging
pip3 install packaging ninja
pip3 install -e '.[flash-attn,deepspeed]'
```

@@ -683,7 +683,7 @@ Please run below to setup env
git clone https://github.com/OpenAccess-AI-Collective/axolotl
cd axolotl
pip3 install packaging
pip3 install packaging ninja
pip3 install -e '.[flash-attn,deepspeed]'
pip3 install -r requirements-dev.txt -r requirements-tests.txt

0 comments on commit cae608f

Please sign in to comment.