diff --git a/applications/Chat/examples/requirements.txt b/applications/Chat/examples/requirements.txt
index 5d0f9f927..c8f8ddc6f 100644
--- a/applications/Chat/examples/requirements.txt
+++ b/applications/Chat/examples/requirements.txt
@@ -1,3 +1,3 @@
 pandas>=1.4.1
 sentencepiece
-colossalai==0.3.1
\ No newline at end of file
+git+https://github.com/hpcaitech/ColossalAI.git@ef4b99ebcda823c32e221383ac9365a82697cd5d#egg=colossalai
\ No newline at end of file
diff --git a/applications/Chat/requirements-test.txt b/applications/Chat/requirements-test.txt
index eb1a77875..9770ea6a8 100644
--- a/applications/Chat/requirements-test.txt
+++ b/applications/Chat/requirements-test.txt
@@ -1,2 +1,2 @@
 pytest
-colossalai==0.3.1
\ No newline at end of file
+git+https://github.com/hpcaitech/ColossalAI.git@ef4b99ebcda823c32e221383ac9365a82697cd5d#egg=colossalai
\ No newline at end of file
diff --git a/applications/Chat/requirements.txt b/applications/Chat/requirements.txt
index e5f5ca093..3755f06e9 100644
--- a/applications/Chat/requirements.txt
+++ b/applications/Chat/requirements.txt
@@ -2,7 +2,7 @@ transformers>=4.20.1
 tqdm
 datasets
 loralib
-colossalai==0.3.1
+git+https://github.com/hpcaitech/ColossalAI.git@ef4b99ebcda823c32e221383ac9365a82697cd5d#egg=colossalai
 torch<2.0.0, >=1.12.1
 langchain
 tokenizers