From 84f523a080083da68bcedf86737eef056889f6a5 Mon Sep 17 00:00:00 2001
From: duanjunwen <935724073@qq.com>
Date: Wed, 4 Jun 2025 09:51:31 +0800
Subject: [PATCH] [Hotfix] fix requirsments (#6338)

* [fix] fix colossalai ascend requirments

* [fix] fix colossalai chat requirements

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* [fix] fix requirments

---------

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
---
 applications/ColossalChat/requirements.txt | 6 ++----
 requirements/requirements.txt              | 6 +++---
 2 files changed, 5 insertions(+), 7 deletions(-)

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