github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-02 17:46:42 +00:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source 
* doc/update requirements-test.txt

* update torch-cuda compatibility check
This commit is contained in:
Zhongkai Zhao
2023-12-27 18:03:45 +08:00
committed by GitHub
parent eae01b6740
commit 64519eb830
2 changed files with 2 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
requirements/requirements-test.txt
View File

@@ -1,5 +1,4 @@
diffusers
fbgemm-gpu==0.2.0
pytest
coverage==7.2.3
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
SentencePiece
ninja
flash_attn==2.0.5
flash_attn
datasets
pydantic
ray