github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-07-28 22:10:06 +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 GitHub
parent 180cea709b
commit 84f523a080
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
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 tqdm
datasets==2.14.7 datasets==2.14.7
loralib loralib
colossalai>=0.4.7
torch==2.5.1 torch==2.5.1
langchain langchain
tokenizers tokenizers
@ -15,11 +14,10 @@ packaging
autoflake==2.2.1 autoflake==2.2.1
black==23.9.1 black==23.9.1
tensorboard tensorboard
six==1.16.0 six==1.17.0
datasets datasets
ninja==1.11.1 ninja==1.11.1
sentencepiece==0.1.99 sentencepiece==0.2.0
flash-attn
tiktoken tiktoken
jsonlines jsonlines
math-verify==0.7.0 math-verify==0.7.0

6
requirements/requirements.txt
View File

@ -8,7 +8,7 @@ click
fabric fabric
contexttimer contexttimer
ninja ninja
torch>=2.2.0,<=2.4.1 torch==2.5.1
safetensors safetensors
einops einops
pydantic pydantic
@ -16,11 +16,11 @@ ray
sentencepiece sentencepiece
google google
protobuf protobuf
transformers==4.39.3 transformers==4.47.0
peft>=0.7.1,<=0.13.2 peft>=0.7.1,<=0.13.2
bitsandbytes>=0.39.0 bitsandbytes>=0.39.0
rpyc==6.0.0 rpyc==6.0.0
fastapi fastapi
uvicorn==0.29.0 uvicorn
galore_torch galore_torch
diffusers==0.29.0 diffusers==0.29.0