From 7ce965c7ccc67e0cbfa60e8b23d21c81a0875c8f Mon Sep 17 00:00:00 2001 From: Haofan Wang Date: Thu, 5 Jan 2023 21:16:42 +0800 Subject: [PATCH] Update requirement_colossalai.txt (#2348) --- examples/images/dreambooth/requirement_colossalai.txt | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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