From d651fb73da9400f3ae0d8e89c1d6bc6f2eda1658 Mon Sep 17 00:00:00 2001
From: Ian Bull <irbull@eclipsesource.com>
Date: Tue, 19 Sep 2023 14:33:03 -0700
Subject: [PATCH] Add deepspeed installation to the README

The "Quickstart" section of the README did not include the installation
of deepspeed. This is required when setting up a fresh instance of
Axolotl. This change-set adds `pip3 install deepspeed` to installation
instructions.

This change-set also updates the requirements.txt file to include
deepspeed.
---
 README.md        | 1 +
 requirements.txt | 1 +
 2 files changed, 2 insertions(+)

diff --git a/README.md b/README.md
index c9b935f53d..6f8b5837b5 100644
--- a/README.md
+++ b/README.md
@@ -88,6 +88,7 @@ cd axolotl
 pip3 install packaging
 pip3 install -e .[flash-attn]
 pip3 install -U git+https://github.com/huggingface/peft.git
+pip3 install deepspeed
 
 # finetune lora
 accelerate launch -m axolotl.cli.train examples/openllama-3b/lora.yml
diff --git a/requirements.txt b/requirements.txt
index 4ef9f5fd2d..796d4c36f9 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -31,3 +31,4 @@ scikit-learn==1.2.2
 pynvml
 art
 wandb
+deepspeed