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

fix README

Browse Source
This commit is contained in:
Fazzie
2023-01-31 10:00:37 +08:00
parent b55deb0662
commit f35326881c
7 changed files with 33 additions and 131 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -1,6 +1,5 @@
# ColoDiffusion: Stable Diffusion with Colossal-AI
Acceleration of AIGC (AI-Generated Content) models such as [Stable Diffusion v1](https://github.com/CompVis/stable-diffusion) and [Stable Diffusion v2](https://github.com/Stability-AI/stablediffusion).
<p id="diffusion_train" align="center">
@@ -57,14 +56,19 @@ pip install transformers==4.19.2 diffusers invisible-watermark
pip install -e .
```
##### Step 2: install lightning
#### Step 2: install lightning
Install Lightning version later than 2022.01.04. We suggest you install lightning from source.
https://github.com/Lightning-AI/lightning.git
```
git clone https://github.com/Lightning-AI/lightning.git
pip install -r requirements.txt
python setup.py install
```
#### Step 3:Install [Colossal-AI](https://colossalai.org/download/) From Our Official Website
##### Step 3:Install [Colossal-AI](https://colossalai.org/download/) From Our Official Website
##### From pip
For example, you can install v0.1.12 from our official website.
@@ -72,6 +76,16 @@ For example, you can install v0.1.12 from our official website.
pip install colossalai==0.1.12+torch1.12cu11.3 -f https://release.colossalai.org
```
##### From source
```
git clone https://github.com/hpcaitech/ColossalAI.git
cd ColossalAI
# install colossalai
CUDA_EXT=1 pip install .
```
### 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.
@@ -122,6 +136,12 @@ It is important for you to configure your volume mapping in order to get the bes
## Download the model checkpoint from pretrained
### stable-diffusion-v2-base
```
wget https://huggingface.co/stabilityai/stable-diffusion-2-base/resolve/main/512-base-ema.ckpt
```
### stable-diffusion-v1-4
Our default model config use the weight from [CompVis/stable-diffusion-v1-4](https://huggingface.co/CompVis/stable-diffusion-v1-4?text=A+mecha+robot+in+a+favela+in+expressionist+style)