diff --git a/examples/images/vit/README.md b/examples/images/vit/README.md
index f78c037ef..4423d85d1 100644
--- a/examples/images/vit/README.md
+++ b/examples/images/vit/README.md
@@ -13,7 +13,7 @@ We use model **vision_transformer** from timm [Link](https://github.com/rwightma
 
 # Requirement
 
-You should install colossalai from main branch with commit 561e904.
+Install colossalai version >= 0.1.11
 
 ## Unit test
 To run unit test, you should install pytest, transformers with: