From 2c5cd0defe110cb1a5c699852f4b38284a2b86b4 Mon Sep 17 00:00:00 2001 From: Woosuk Kwon Date: Sat, 1 Apr 2023 19:00:20 -0700 Subject: [PATCH] Add ninja to dependency (#21) --- README.md | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/README.md b/README.md index f9bf1ac2f460..8df3fecec4b5 100644 --- a/README.md +++ b/README.md @@ -6,7 +6,8 @@ pip install psutil numpy ray torch pip install git+https://github.com/huggingface/transformers # Required for LLaMA. pip install sentencepiece # Required for LlamaTokenizer. -pip install flash-attn # This may take up to 20 mins. +pip install ninja # To parallelize the compilation of flash-attn. +pip install flash-attn # This may take up to 10 mins. pip install -e . ```