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

[misc] update requirements

pull/4445/head
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
SentencePiece
ninja
flash_attn>=2.0
flash_attn==2.0.5
datasets
ninja
flash-attn>=2.0

3
requirements/requirements.txt
View File

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