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

[doc] Update required third-party library list for testing and torch comptibility checking (#5207) 

* doc/update requirements-test.txt

* update torch-cuda compatibility check
pull/5217/head
Zhongkai Zhao 2023-12-27 18:03:45 +08:00 committed by GitHub
parent eae01b6740
commit 64519eb830
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 2 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
.compatibility
View File

@ -1,3 +1,2 @@
1.12.0-11.3.0
1.13.0-11.6.0
2.0.0-11.7.0 2.0.0-11.7.0
2.1.0-11.8.0

3
requirements/requirements-test.txt
View File

@ -1,5 +1,4 @@
diffusers diffusers
fbgemm-gpu==0.2.0
pytest pytest
coverage==7.2.3 coverage==7.2.3
git+https://github.com/hpcaitech/pytest-testmon git+https://github.com/hpcaitech/pytest-testmon
@ -16,7 +15,7 @@ triton==2.1.0
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.5 flash_attn
datasets datasets
pydantic pydantic
ray ray