[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 .