Skip to content

Commit 2c5cd0d

Browse files
authored
Add ninja to dependency (#21)
1 parent a90c97d commit 2c5cd0d

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

README.md

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,8 @@
66
pip install psutil numpy ray torch
77
pip install git+https://github.com/huggingface/transformers # Required for LLaMA.
88
pip install sentencepiece # Required for LlamaTokenizer.
9-
pip install flash-attn # This may take up to 20 mins.
9+
pip install ninja # To parallelize the compilation of flash-attn.
10+
pip install flash-attn # This may take up to 10 mins.
1011
pip install -e .
1112
```
1213

0 commit comments

Comments
 (0)