github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-16 22:52:25 +00:00
Code Issues Packages Projects Releases Wiki Activity

add ci (#2641)

Browse Source
This commit is contained in:
Fazzie-Maqianli
2023-02-15 09:55:53 +08:00
committed by GitHub
parent 0b2a738393
commit d03f4429c1
3 changed files with 25 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
examples/images/diffusion/README.md
View File

@@ -92,6 +92,12 @@ cd ColossalAI
CUDA_EXT=1 pip install .
```
#### Step 3:Accelerate with flash attention by xformers(Optional)
```
pip install xformers
```
### Option #2: Use Docker
To use the stable diffusion Docker image, you can either build using the provided the [Dockerfile](./docker/Dockerfile) or pull a Docker image from our Docker hub.