[example] add example requirement (#2345)

2025-09-12 12:47:21 +00:00 · 2023-01-06 09:03:29 +08:00
parent 7ce965c7cc
commit d7352bef2c
14 changed files with 22 additions and 0 deletions
--- a/examples/images/diffusion/environment.yaml
+++ b/examples/images/diffusion/environment.yaml
@@ -27,4 +27,5 @@ dependencies:
    - torchmetrics==0.7.0
    - prefetch_generator
    - datasets
+    - colossalai
    - -e .
--- a/examples/images/diffusion/requirements.txt
+++ b/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 .
--- a/examples/images/dreambooth/requirements.txt
+++ b/examples/images/dreambooth/requirements.txt
@@ -5,3 +5,4 @@ transformers>=4.21.0
 ftfy
 tensorboard
 modelcards
+colossalai
--- a/examples/images/vit/requirements.txt
+++ b/examples/images/vit/requirements.txt
@@ -0,0 +1,2 @@
+colossalai >= 0.1.12
+torch >= 1.8.1