diff --git a/README.md b/README.md index 5d9a29a83..953bc0be5 100644 --- a/README.md +++ b/README.md @@ -55,6 +55,7 @@ Features: ### Installation ```bash +pip3 install -U packaging setuptools wheel ninja pip3 install --no-build-isolation axolotl[flash-attn,deepspeed] # Download example axolotl configs, deepspeed configs diff --git a/docs/installation.qmd b/docs/installation.qmd index 95f15e78e..3cff0bd32 100644 --- a/docs/installation.qmd +++ b/docs/installation.qmd @@ -22,6 +22,7 @@ This guide covers all the ways you can install and set up Axolotl for your envir ### PyPI Installation (Recommended) {#sec-pypi} ```{.bash} +pip3 install -U packaging setuptools wheel ninja pip3 install --no-build-isolation axolotl[flash-attn,deepspeed] ``` @@ -37,7 +38,7 @@ For the latest features between releases: ```{.bash} git clone https://github.com/axolotl-ai-cloud/axolotl.git cd axolotl -pip3 install packaging ninja +pip3 install -U packaging setuptools wheel ninja pip3 install --no-build-isolation -e '.[flash-attn,deepspeed]' ``` @@ -107,7 +108,7 @@ We recommend using WSL2 (Windows Subsystem for Linux) or Docker. 2. Install PyTorch: https://pytorch.org/get-started/locally/ 3. Install Axolotl: ```{.bash} - pip3 install packaging + pip3 install -U packaging setuptools wheel ninja pip3 install --no-build-isolation -e '.[flash-attn,deepspeed]' ``` 4. (Optional) Login to Hugging Face: