github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-07-30 23:06:01 +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
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