diff --git a/examples/images/dreambooth/requirement_colossalai.txt b/examples/images/dreambooth/requirement_colossalai.txt
index 2591a2726..c4a0e9170 100644
--- a/examples/images/dreambooth/requirement_colossalai.txt
+++ b/examples/images/dreambooth/requirement_colossalai.txt
@@ -5,4 +5,4 @@ ftfy
 tensorboard
 modelcards
 transformers
-colossalai==0.1.11rc5+torch1.12cu11.3 -f https://release.colossalai.org
+colossalai==0.2.0+torch1.12cu11.3 -f https://release.colossalai.org