[example] add example requirement (#2345)

examples/images/diffusion/environment.yaml

@@ -27,4 +27,5 @@ dependencies:
     - torchmetrics==0.7.0
     - prefetch_generator
     - datasets
+    - colossalai
     - -e .

examples/images/diffusion/requirements.txt

@@ -14,4 +14,5 @@ webdataset==0.2.5
 open-clip-torch==2.7.0
 gradio==3.11
 datasets
+colossalai
 -e .

examples/images/dreambooth/requirements.txt

@@ -5,3 +5,4 @@ transformers>=4.21.0
 ftfy
 tensorboard
 modelcards
+colossalai

examples/images/vit/requirements.txt

@@ -0,0 +1,2 @@
+colossalai >= 0.1.12
+torch >= 1.8.1

examples/language/palm/requirements.txt

@@ -0,0 +1,2 @@
+colossalai >= 0.1.12
+torch >= 1.8.1

examples/language/roberta/requirements.txt

@@ -0,0 +1,2 @@
+colossalai >= 0.1.12
+torch >= 1.8.1

examples/tutorial/auto_parallel/requirements.txt

@@ -0,0 +1,2 @@
+colossalai >= 0.1.12
+torch >= 1.8.1

examples/tutorial/hybrid_parallel/requirements.txt

@@ -0,0 +1,2 @@
+colossalai >= 0.1.12
+torch >= 1.8.1

examples/tutorial/large_batch_optimizer/requirements.txt

@@ -0,0 +1,2 @@
+colossalai >= 0.1.12
+torch >= 1.8.1

examples/tutorial/opt/inference/requirements.txt

@@ -6,3 +6,4 @@ sanic_ext==22.9.0
 torch>=1.10.0
 transformers==4.23.1
 uvicorn==0.19.0
+colossalai

examples/tutorial/requirements.txt

@@ -0,0 +1,2 @@
+colossalai >= 0.1.12
+torch >= 1.8.1

examples/tutorial/sequence_parallel/requirements.txt

@@ -0,0 +1,2 @@
+colossalai >= 0.1.12
+torch >= 1.8.1

examples/tutorial/stable_diffusion/environment.yaml

@@ -28,6 +28,7 @@ dependencies:
     - torchmetrics==0.7.0
     - kornia==0.6
     - prefetch_generator
+    - colossalai
     - -e git+https://github.com/CompVis/taming-transformers.git@master#egg=taming-transformers
     - -e git+https://github.com/openai/CLIP.git@main#egg=clip
     - -e .

examples/tutorial/stable_diffusion/requirements.txt

@@ -16,6 +16,7 @@ torchmetrics==0.6.0
 kornia==0.6
 opencv-python==4.6.0.66
 prefetch_generator
+colossalai
 -e git+https://github.com/CompVis/taming-transformers.git@master#egg=taming-transformers
 -e git+https://github.com/openai/CLIP.git@main#egg=clip
 -e .