diff --git a/applications/ColossalChat/examples/requirements.txt b/applications/ColossalChat/examples/requirements.txt
index 838590f4b..91f25a5cf 100644
--- a/applications/ColossalChat/examples/requirements.txt
+++ b/applications/ColossalChat/examples/requirements.txt
@@ -1,4 +1,4 @@
 pandas>=1.4.1
 sentencepiece
-colossalai
+colossalai==0.4.0
 prompt_toolkit
diff --git a/applications/ColossalChat/requirements.txt b/applications/ColossalChat/requirements.txt
index 639ff6aef..acf4c64bf 100755
--- a/applications/ColossalChat/requirements.txt
+++ b/applications/ColossalChat/requirements.txt
@@ -2,8 +2,8 @@ transformers==4.36.2
 tqdm
 datasets==2.14.7
 loralib
-colossalai>=0.3.7
-torch>=1.12.1
+colossalai==0.4.0
+torch>=2.1.0
 langchain
 tokenizers
 fastapi