github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-08-07 02:54:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

[misc] update requirements

Browse Source
This commit is contained in:
ver217 2023-08-15 17:07:29 +08:00 committed by Hongxin Liu
parent 172f7fa3cf
commit 922302263b
2 changed files with 2 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
requirements/requirements-test.txt
View File

@ -16,7 +16,5 @@ triton==2.0.0.dev20221202
requests==2.27.1 # downgrade to avoid huggingface error https://github.com/huggingface/transformers/issues/17611 requests==2.27.1 # downgrade to avoid huggingface error https://github.com/huggingface/transformers/issues/17611
SentencePiece SentencePiece
ninja ninja
flash_attn>=2.0 flash_attn==2.0.5
datasets datasets
ninja
flash-attn>=2.0

3
requirements/requirements.txt
View File

@ -8,7 +8,6 @@ click
fabric fabric
contexttimer contexttimer
ninja ninja
torch>=1.11 torch>=1.12
safetensors safetensors
flash_attn>=2.0
einops einops