Skip to content

Commit 84f523a

Browse files
[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>
1 parent 180cea7 commit 84f523a

File tree

2 files changed

+5
-7
lines changed

2 files changed

+5
-7
lines changed

applications/ColossalChat/requirements.txt

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,6 @@ transformers==4.47.0
22
tqdm
33
datasets==2.14.7
44
loralib
5-
colossalai>=0.4.7
65
torch==2.5.1
76
langchain
87
tokenizers
@@ -15,11 +14,10 @@ packaging
1514
autoflake==2.2.1
1615
black==23.9.1
1716
tensorboard
18-
six==1.16.0
17+
six==1.17.0
1918
datasets
2019
ninja==1.11.1
21-
sentencepiece==0.1.99
22-
flash-attn
20+
sentencepiece==0.2.0
2321
tiktoken
2422
jsonlines
2523
math-verify==0.7.0

requirements/requirements.txt

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -8,19 +8,19 @@ click
88
fabric
99
contexttimer
1010
ninja
11-
torch>=2.2.0,<=2.4.1
11+
torch==2.5.1
1212
safetensors
1313
einops
1414
pydantic
1515
ray
1616
sentencepiece
1717
google
1818
protobuf
19-
transformers==4.39.3
19+
transformers==4.47.0
2020
peft>=0.7.1,<=0.13.2
2121
bitsandbytes>=0.39.0
2222
rpyc==6.0.0
2323
fastapi
24-
uvicorn==0.29.0
24+
uvicorn
2525
galore_torch
2626
diffusers==0.29.0

0 commit comments

Comments
 (0)