From cae608f587e7b32ae5e478547d76f8236dae17c5 Mon Sep 17 00:00:00 2001 From: James Melvin Ebenezer Date: Tue, 2 Apr 2024 14:06:41 +0530 Subject: [PATCH] Added pip install ninja to accelerate installation of flash-attn (#1461) * Added pip install ninja to accelerate installation of flash-attn * doc: cleanup --- README.md | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/README.md b/README.md index ad5fa379a..b4b7b1e67 100644 --- a/README.md +++ b/README.md @@ -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