github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

[misc] update requirements (#5787)

pull/5788/head
Hongxin Liu 2024-06-06 15:55:34 +08:00 committed by GitHub
parent a1e39f4c0d
commit 5ead00ffc5
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
requirements/requirements.txt
View File

@ -8,7 +8,7 @@ click
fabric
contexttimer
ninja
torch>=2.1.0
torch>=2.1.0,<2.3.0
safetensors
einops
pydantic
@ -16,7 +16,7 @@ ray
sentencepiece
google
protobuf
transformers==4.36.2
transformers>=4.36.2,<4.40.0
peft>=0.7.1
bitsandbytes>=0.39.0
rpyc==6.0.0