diff --git a/applications/ColossalChat/requirements.txt b/applications/ColossalChat/requirements.txt
index e1b8291ab..df579f2a7 100755
--- a/applications/ColossalChat/requirements.txt
+++ b/applications/ColossalChat/requirements.txt
@@ -2,7 +2,6 @@ transformers==4.47.0
 tqdm
 datasets==2.14.7
 loralib
-colossalai>=0.4.7
 torch==2.5.1
 langchain
 tokenizers
@@ -15,11 +14,10 @@ packaging
 autoflake==2.2.1
 black==23.9.1
 tensorboard
-six==1.16.0
+six==1.17.0
 datasets
 ninja==1.11.1
-sentencepiece==0.1.99
-flash-attn
+sentencepiece==0.2.0
 tiktoken
 jsonlines
 math-verify==0.7.0
diff --git a/requirements/requirements.txt b/requirements/requirements.txt
index f357c45fd..9c110a1f4 100644
--- a/requirements/requirements.txt
+++ b/requirements/requirements.txt
@@ -8,7 +8,7 @@ click
 fabric
 contexttimer
 ninja
-torch>=2.2.0,<=2.4.1
+torch==2.5.1
 safetensors
 einops
 pydantic
@@ -16,11 +16,11 @@ ray
 sentencepiece
 google
 protobuf
-transformers==4.39.3
+transformers==4.47.0
 peft>=0.7.1,<=0.13.2
 bitsandbytes>=0.39.0
 rpyc==6.0.0
 fastapi
-uvicorn==0.29.0
+uvicorn
 galore_torch
 diffusers==0.29.0