From 284b432ffd4d53a3ff313758d7e4bc978f4e74ba Mon Sep 17 00:00:00 2001 From: Lincoln Stein Date: Wed, 18 Jan 2023 22:34:36 -0500 Subject: [PATCH] add triton install instructions --- docs/installation/070_INSTALL_XFORMERS.md | 13 +++++++++++-- 1 file changed, 11 insertions(+), 2 deletions(-) diff --git a/docs/installation/070_INSTALL_XFORMERS.md b/docs/installation/070_INSTALL_XFORMERS.md index a406c28c22..99138744f8 100644 --- a/docs/installation/070_INSTALL_XFORMERS.md +++ b/docs/installation/070_INSTALL_XFORMERS.md @@ -73,7 +73,16 @@ most up to date libraries with this command: pip install --upgrade --force-reinstall torch torchvision ``` -### 3. Install source code build prerequisites +### 3. Install the triton module + +This module isn't necessary for xFormers image inference optimization, +but avoids a startup warning. + +```sh +pip install triton +``` + +### 4. Install source code build prerequisites To build xFormers from source, you will need the `build-essentials` package. If you don't have it installed already, run: @@ -82,7 +91,7 @@ package. If you don't have it installed already, run: sudo apt install build-essential ``` -### 4. Build xFormers +### 5. Build xFormers There is no pip wheel package for xFormers at this time (January 2023). Although there is a conda package, InvokeAI no longer