From 40aa162808d72d3a0340fe8484138eade88b3628 Mon Sep 17 00:00:00 2001 From: Haofan Wang Date: Thu, 12 Jan 2023 18:34:10 +0800 Subject: [PATCH] [Docs] Update README.md (#1960) Update README.md --- examples/research_projects/colossalai/README.md | 12 ++++++++---- 1 file changed, 8 insertions(+), 4 deletions(-) diff --git a/examples/research_projects/colossalai/README.md b/examples/research_projects/colossalai/README.md index a306a3abfc..7c428bbce7 100644 --- a/examples/research_projects/colossalai/README.md +++ b/examples/research_projects/colossalai/README.md @@ -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