github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-07-24 12:09:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

[Hotfix] fix requirsments (#6338)

Browse Source 
* [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>
This commit is contained in:
duanjunwen 2025-06-04 09:51:31 +08:00 committed by xysheng-colossal
parent 2783ddecf5
commit 1304be44ae
2 changed files with 5 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
applications/ColossalChat/requirements.txt
View File

@ -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

6
requirements/requirements.txt
View File

@ -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