Browse Source

[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 11 months ago committed by GitHub
parent
commit
64519eb830
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
  1. 3
      .compatibility
  2. 3
      requirements/requirements-test.txt

3
.compatibility

@ -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

@ -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

Loading…
Cancel
Save