add triton install instructions

This commit is contained in:
Lincoln Stein 2023-01-18 22:34:36 -05:00
parent 4556f29359
commit 284b432ffd

View File

@ -73,7 +73,16 @@ most up to date libraries with this command:
pip install --upgrade --force-reinstall torch torchvision 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` To build xFormers from source, you will need the `build-essentials`
package. If you don't have it installed already, run: 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 sudo apt install build-essential
``` ```
### 4. Build xFormers ### 5. Build xFormers
There is no pip wheel package for xFormers at this time (January There is no pip wheel package for xFormers at this time (January
2023). Although there is a conda package, InvokeAI no longer 2023). Although there is a conda package, InvokeAI no longer