mirror of https://github.com/hpcaitech/ColossalAI
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
21 lines
545 B
21 lines
545 B
diffusers
|
|
fbgemm-gpu==0.2.0
|
|
pytest
|
|
coverage==7.2.3
|
|
git+https://github.com/hpcaitech/pytest-testmon
|
|
torchvision
|
|
transformers==4.33.0
|
|
timm
|
|
titans
|
|
torchaudio
|
|
torchx-nightly==2022.6.29 # torchrec 0.2.0 requires torchx-nightly. This package is updated every day. We fix the version to a specific date to avoid breaking changes.
|
|
torchrec==0.2.0
|
|
contexttimer
|
|
einops
|
|
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.5
|
|
datasets
|