1
0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00

[Docs] Update README.md (#1960)

Update README.md
This commit is contained in:
Haofan Wang
2023-01-12 18:34:10 +08:00
committed by GitHub
parent f06e4e5579
commit 40aa162808

View File

@@ -10,20 +10,24 @@ By accommodating model data in CPU and GPU and moving the data to the computing
Before running the scripts, make sure to install the library's training dependencies:
```bash
pip install -r requirements_colossalai.txt
pip install -r requirements.txt
```
### Install [colossalai](https://github.com/hpcaitech/ColossalAI.git)
## Install [ColossalAI](https://github.com/hpcaitech/ColossalAI.git)
**From PyPI**
```bash
pip install colossalai==0.2.0+torch1.12cu11.3 -f https://release.colossalai.org
pip install colossalai
```
**From source**
```bash
git clone https://github.com/hpcaitech/ColossalAI.git
python setup.py install
cd ColossalAI
# install colossalai
pip install .
```
## Dataset for Teyvat BLIP captions