github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-17 07:00:37 +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
1 changed files with 2 additions and 2 deletions
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