github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-06-22 21:49:08 +00:00
Code Issues Packages Projects Releases Wiki Activity

[misc] update requirements (#5787)

Browse Source
This commit is contained in:
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 fabric
contexttimer contexttimer
ninja ninja
torch>=2.1.0 torch>=2.1.0,<2.3.0
safetensors safetensors
einops einops
pydantic pydantic
@ -16,7 +16,7 @@ ray
sentencepiece sentencepiece
google google
protobuf protobuf
transformers==4.36.2 transformers>=4.36.2,<4.40.0
peft>=0.7.1 peft>=0.7.1
bitsandbytes>=0.39.0 bitsandbytes>=0.39.0
rpyc==6.0.0 rpyc==6.0.0